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 <stdlib.h>
18#include <sys/socket.h>
19#include <sys/types.h>
20#include <netinet/in.h>
21#include <arpa/inet.h>
22#include <dirent.h>
23#include <errno.h>
24#include <fcntl.h>
25#include <fs_mgr.h>
26#include <stdio.h>
27#include <string.h>
28#include <stdint.h>
29#include <inttypes.h>
30
31#define LOG_TAG "VoldCmdListener"
32
33#include <base/stringprintf.h>
34#include <cutils/fs.h>
35#include <cutils/log.h>
36
37#include <sysutils/SocketClient.h>
38#include <private/android_filesystem_config.h>
39
40#include "CommandListener.h"
41#include "VolumeManager.h"
42#include "VolumeBase.h"
43#include "ResponseCode.h"
44#include "Process.h"
45#include "Loop.h"
46#include "Devmapper.h"
47#include "cryptfs.h"
48#include "MoveTask.h"
49#include "TrimTask.h"
50
51#define DUMP_ARGS 0
52
53CommandListener::CommandListener() :
54                 FrameworkListener("vold", true) {
55    registerCmd(new DumpCmd());
56    registerCmd(new VolumeCmd());
57    registerCmd(new AsecCmd());
58    registerCmd(new ObbCmd());
59    registerCmd(new StorageCmd());
60    registerCmd(new FstrimCmd());
61}
62
63#if DUMP_ARGS
64void CommandListener::dumpArgs(int argc, char **argv, int argObscure) {
65    char buffer[4096];
66    char *p = buffer;
67
68    memset(buffer, 0, sizeof(buffer));
69    int i;
70    for (i = 0; i < argc; i++) {
71        unsigned int len = strlen(argv[i]) + 1; // Account for space
72        if (i == argObscure) {
73            len += 2; // Account for {}
74        }
75        if (((p - buffer) + len) < (sizeof(buffer)-1)) {
76            if (i == argObscure) {
77                *p++ = '{';
78                *p++ = '}';
79                *p++ = ' ';
80                continue;
81            }
82            strcpy(p, argv[i]);
83            p+= strlen(argv[i]);
84            if (i != (argc -1)) {
85                *p++ = ' ';
86            }
87        }
88    }
89    SLOGD("%s", buffer);
90}
91#else
92void CommandListener::dumpArgs(int /*argc*/, char ** /*argv*/, int /*argObscure*/) { }
93#endif
94
95int CommandListener::sendGenericOkFail(SocketClient *cli, int cond) {
96    if (!cond) {
97        return cli->sendMsg(ResponseCode::CommandOkay, "Command succeeded", false);
98    } else {
99        return cli->sendMsg(ResponseCode::OperationFailed, "Command failed", false);
100    }
101}
102
103CommandListener::DumpCmd::DumpCmd() :
104                 VoldCommand("dump") {
105}
106
107int CommandListener::DumpCmd::runCommand(SocketClient *cli,
108                                         int /*argc*/, char ** /*argv*/) {
109    cli->sendMsg(0, "Dumping loop status", false);
110    if (Loop::dumpState(cli)) {
111        cli->sendMsg(ResponseCode::CommandOkay, "Loop dump failed", true);
112    }
113    cli->sendMsg(0, "Dumping DM status", false);
114    if (Devmapper::dumpState(cli)) {
115        cli->sendMsg(ResponseCode::CommandOkay, "Devmapper dump failed", true);
116    }
117    cli->sendMsg(0, "Dumping mounted filesystems", false);
118    FILE *fp = fopen("/proc/mounts", "r");
119    if (fp) {
120        char line[1024];
121        while (fgets(line, sizeof(line), fp)) {
122            line[strlen(line)-1] = '\0';
123            cli->sendMsg(0, line, false);;
124        }
125        fclose(fp);
126    }
127
128    cli->sendMsg(ResponseCode::CommandOkay, "dump complete", false);
129    return 0;
130}
131
132CommandListener::VolumeCmd::VolumeCmd() :
133                 VoldCommand("volume") {
134}
135
136int CommandListener::VolumeCmd::runCommand(SocketClient *cli,
137                                           int argc, char **argv) {
138    dumpArgs(argc, argv, -1);
139
140    if (argc < 2) {
141        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
142        return 0;
143    }
144
145    VolumeManager *vm = VolumeManager::Instance();
146    std::lock_guard<std::mutex> lock(vm->getLock());
147
148    // TODO: tease out methods not directly related to volumes
149
150    std::string cmd(argv[1]);
151    if (cmd == "reset") {
152        return sendGenericOkFail(cli, vm->reset());
153
154    } else if (cmd == "shutdown") {
155        return sendGenericOkFail(cli, vm->shutdown());
156
157    } else if (cmd == "debug") {
158        return sendGenericOkFail(cli, vm->setDebug(true));
159
160    } else if (cmd == "partition" && argc > 3) {
161        // partition [diskId] [public|private|mixed] [ratio]
162        std::string id(argv[2]);
163        auto disk = vm->findDisk(id);
164        if (disk == nullptr) {
165            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown disk", false);
166        }
167
168        std::string type(argv[3]);
169        if (type == "public") {
170            return sendGenericOkFail(cli, disk->partitionPublic());
171        } else if (type == "private") {
172            return sendGenericOkFail(cli, disk->partitionPrivate());
173        } else if (type == "mixed") {
174            if (argc < 4) {
175                return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
176            }
177            int frac = atoi(argv[4]);
178            return sendGenericOkFail(cli, disk->partitionMixed(frac));
179        } else {
180            return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
181        }
182
183    } else if (cmd == "mkdirs" && argc > 2) {
184        // mkdirs [path]
185        return sendGenericOkFail(cli, vm->mkdirs(argv[2]));
186
187    } else if (cmd == "user_added" && argc > 3) {
188        // user_added [user] [serial]
189        return sendGenericOkFail(cli, vm->onUserAdded(atoi(argv[2]), atoi(argv[3])));
190
191    } else if (cmd == "user_removed" && argc > 2) {
192        // user_removed [user]
193        return sendGenericOkFail(cli, vm->onUserRemoved(atoi(argv[2])));
194
195    } else if (cmd == "user_started" && argc > 2) {
196        // user_started [user]
197        return sendGenericOkFail(cli, vm->onUserStarted(atoi(argv[2])));
198
199    } else if (cmd == "user_stopped" && argc > 2) {
200        // user_stopped [user]
201        return sendGenericOkFail(cli, vm->onUserStopped(atoi(argv[2])));
202
203    } else if (cmd == "mount" && argc > 2) {
204        // mount [volId] [flags] [user]
205        std::string id(argv[2]);
206        auto vol = vm->findVolume(id);
207        if (vol == nullptr) {
208            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
209        }
210
211        int mountFlags = (argc > 3) ? atoi(argv[3]) : 0;
212        userid_t mountUserId = (argc > 4) ? atoi(argv[4]) : -1;
213
214        vol->setMountFlags(mountFlags);
215        vol->setMountUserId(mountUserId);
216
217        int res = vol->mount();
218        if (mountFlags & android::vold::VolumeBase::MountFlags::kPrimary) {
219            vm->setPrimary(vol);
220        }
221        return sendGenericOkFail(cli, res);
222
223    } else if (cmd == "unmount" && argc > 2) {
224        // unmount [volId]
225        std::string id(argv[2]);
226        auto vol = vm->findVolume(id);
227        if (vol == nullptr) {
228            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
229        }
230
231        return sendGenericOkFail(cli, vol->unmount());
232
233    } else if (cmd == "format" && argc > 3) {
234        // format [volId] [fsType|auto]
235        std::string id(argv[2]);
236        std::string fsType(argv[3]);
237        auto vol = vm->findVolume(id);
238        if (vol == nullptr) {
239            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
240        }
241
242        return sendGenericOkFail(cli, vol->format(fsType));
243
244    } else if (cmd == "move_storage" && argc > 3) {
245        // move_storage [fromVolId] [toVolId]
246        auto fromVol = vm->findVolume(std::string(argv[2]));
247        auto toVol = vm->findVolume(std::string(argv[3]));
248        if (fromVol == nullptr || toVol == nullptr) {
249            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
250        }
251
252        (new android::vold::MoveTask(fromVol, toVol))->start();
253        return sendGenericOkFail(cli, 0);
254
255    } else if (cmd == "benchmark" && argc > 2) {
256        // benchmark [volId]
257        std::string id(argv[2]);
258        nsecs_t res = vm->benchmarkPrivate(id);
259        return cli->sendMsg(ResponseCode::CommandOkay,
260                android::base::StringPrintf("%" PRId64, res).c_str(), false);
261
262    } else if (cmd == "forget_partition" && argc > 2) {
263        // forget_partition [partGuid]
264        std::string partGuid(argv[2]);
265        return sendGenericOkFail(cli, vm->forgetPartition(partGuid));
266
267    } else if (cmd == "remount_uid" && argc > 3) {
268        // remount_uid [uid] [none|default|read|write]
269        uid_t uid = atoi(argv[2]);
270        std::string mode(argv[3]);
271        return sendGenericOkFail(cli, vm->remountUid(uid, mode));
272    }
273
274    return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
275}
276
277CommandListener::StorageCmd::StorageCmd() :
278                 VoldCommand("storage") {
279}
280
281int CommandListener::StorageCmd::runCommand(SocketClient *cli,
282                                                      int argc, char **argv) {
283    /* Guarantied to be initialized by vold's main() before the CommandListener is active */
284    extern struct fstab *fstab;
285
286    dumpArgs(argc, argv, -1);
287
288    if (argc < 2) {
289        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
290        return 0;
291    }
292
293    if (!strcmp(argv[1], "mountall")) {
294        if (argc != 2) {
295            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: mountall", false);
296            return 0;
297        }
298        fs_mgr_mount_all(fstab);
299        cli->sendMsg(ResponseCode::CommandOkay, "Mountall ran successfully", false);
300        return 0;
301    }
302    if (!strcmp(argv[1], "users")) {
303        DIR *dir;
304        struct dirent *de;
305
306        if (argc < 3) {
307            cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument: user <mountpoint>", false);
308            return 0;
309        }
310        if (!(dir = opendir("/proc"))) {
311            cli->sendMsg(ResponseCode::OperationFailed, "Failed to open /proc", true);
312            return 0;
313        }
314
315        while ((de = readdir(dir))) {
316            int pid = Process::getPid(de->d_name);
317
318            if (pid < 0) {
319                continue;
320            }
321
322            char processName[255];
323            Process::getProcessName(pid, processName, sizeof(processName));
324
325            if (Process::checkFileDescriptorSymLinks(pid, argv[2]) ||
326                Process::checkFileMaps(pid, argv[2]) ||
327                Process::checkSymLink(pid, argv[2], "cwd") ||
328                Process::checkSymLink(pid, argv[2], "root") ||
329                Process::checkSymLink(pid, argv[2], "exe")) {
330
331                char msg[1024];
332                snprintf(msg, sizeof(msg), "%d %s", pid, processName);
333                cli->sendMsg(ResponseCode::StorageUsersListResult, msg, false);
334            }
335        }
336        closedir(dir);
337        cli->sendMsg(ResponseCode::CommandOkay, "Storage user list complete", false);
338    } else {
339        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown storage cmd", false);
340    }
341    return 0;
342}
343
344CommandListener::AsecCmd::AsecCmd() :
345                 VoldCommand("asec") {
346}
347
348void CommandListener::AsecCmd::listAsecsInDirectory(SocketClient *cli, const char *directory) {
349    DIR *d = opendir(directory);
350
351    if (!d) {
352        cli->sendMsg(ResponseCode::OperationFailed, "Failed to open asec dir", true);
353        return;
354    }
355
356    size_t dirent_len = offsetof(struct dirent, d_name) +
357            fpathconf(dirfd(d), _PC_NAME_MAX) + 1;
358
359    struct dirent *dent = (struct dirent *) malloc(dirent_len);
360    if (dent == NULL) {
361        cli->sendMsg(ResponseCode::OperationFailed, "Failed to allocate memory", true);
362        return;
363    }
364
365    struct dirent *result;
366
367    while (!readdir_r(d, dent, &result) && result != NULL) {
368        if (dent->d_name[0] == '.')
369            continue;
370        if (dent->d_type != DT_REG)
371            continue;
372        size_t name_len = strlen(dent->d_name);
373        if (name_len > 5 && name_len < 260 &&
374                !strcmp(&dent->d_name[name_len - 5], ".asec")) {
375            char id[255];
376            memset(id, 0, sizeof(id));
377            strlcpy(id, dent->d_name, name_len - 4);
378            cli->sendMsg(ResponseCode::AsecListResult, id, false);
379        }
380    }
381    closedir(d);
382
383    free(dent);
384}
385
386int CommandListener::AsecCmd::runCommand(SocketClient *cli,
387                                                      int argc, char **argv) {
388    if (argc < 2) {
389        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
390        return 0;
391    }
392
393    VolumeManager *vm = VolumeManager::Instance();
394    int rc = 0;
395
396    if (!strcmp(argv[1], "list")) {
397        dumpArgs(argc, argv, -1);
398
399        listAsecsInDirectory(cli, VolumeManager::SEC_ASECDIR_EXT);
400        listAsecsInDirectory(cli, VolumeManager::SEC_ASECDIR_INT);
401    } else if (!strcmp(argv[1], "create")) {
402        dumpArgs(argc, argv, 5);
403        if (argc != 8) {
404            cli->sendMsg(ResponseCode::CommandSyntaxError,
405                    "Usage: asec create <container-id> <size_mb> <fstype> <key> <ownerUid> "
406                    "<isExternal>", false);
407            return 0;
408        }
409
410        unsigned int numSectors = (atoi(argv[3]) * (1024 * 1024)) / 512;
411        const bool isExternal = (atoi(argv[7]) == 1);
412        rc = vm->createAsec(argv[2], numSectors, argv[4], argv[5], atoi(argv[6]), isExternal);
413    } else if (!strcmp(argv[1], "resize")) {
414        dumpArgs(argc, argv, -1);
415        if (argc != 5) {
416            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec resize <container-id> <size_mb> <key>", false);
417            return 0;
418        }
419        unsigned int numSectors = (atoi(argv[3]) * (1024 * 1024)) / 512;
420        rc = vm->resizeAsec(argv[2], numSectors, argv[4]);
421    } else if (!strcmp(argv[1], "finalize")) {
422        dumpArgs(argc, argv, -1);
423        if (argc != 3) {
424            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec finalize <container-id>", false);
425            return 0;
426        }
427        rc = vm->finalizeAsec(argv[2]);
428    } else if (!strcmp(argv[1], "fixperms")) {
429        dumpArgs(argc, argv, -1);
430        if  (argc != 5) {
431            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fixperms <container-id> <gid> <filename>", false);
432            return 0;
433        }
434
435        char *endptr;
436        gid_t gid = (gid_t) strtoul(argv[3], &endptr, 10);
437        if (*endptr != '\0') {
438            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fixperms <container-id> <gid> <filename>", false);
439            return 0;
440        }
441
442        rc = vm->fixupAsecPermissions(argv[2], gid, argv[4]);
443    } else if (!strcmp(argv[1], "destroy")) {
444        dumpArgs(argc, argv, -1);
445        if (argc < 3) {
446            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec destroy <container-id> [force]", false);
447            return 0;
448        }
449        bool force = false;
450        if (argc > 3 && !strcmp(argv[3], "force")) {
451            force = true;
452        }
453        rc = vm->destroyAsec(argv[2], force);
454    } else if (!strcmp(argv[1], "mount")) {
455        dumpArgs(argc, argv, 3);
456        if (argc != 6) {
457            cli->sendMsg(ResponseCode::CommandSyntaxError,
458                    "Usage: asec mount <namespace-id> <key> <ownerUid> <ro|rw>", false);
459            return 0;
460        }
461        bool readOnly = true;
462        if (!strcmp(argv[5], "rw")) {
463            readOnly = false;
464        }
465        rc = vm->mountAsec(argv[2], argv[3], atoi(argv[4]), readOnly);
466    } else if (!strcmp(argv[1], "unmount")) {
467        dumpArgs(argc, argv, -1);
468        if (argc < 3) {
469            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec unmount <container-id> [force]", false);
470            return 0;
471        }
472        bool force = false;
473        if (argc > 3 && !strcmp(argv[3], "force")) {
474            force = true;
475        }
476        rc = vm->unmountAsec(argv[2], force);
477    } else if (!strcmp(argv[1], "rename")) {
478        dumpArgs(argc, argv, -1);
479        if (argc != 4) {
480            cli->sendMsg(ResponseCode::CommandSyntaxError,
481                    "Usage: asec rename <old_id> <new_id>", false);
482            return 0;
483        }
484        rc = vm->renameAsec(argv[2], argv[3]);
485    } else if (!strcmp(argv[1], "path")) {
486        dumpArgs(argc, argv, -1);
487        if (argc != 3) {
488            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec path <container-id>", false);
489            return 0;
490        }
491        char path[255];
492
493        if (!(rc = vm->getAsecMountPath(argv[2], path, sizeof(path)))) {
494            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
495            return 0;
496        }
497    } else if (!strcmp(argv[1], "fspath")) {
498        dumpArgs(argc, argv, -1);
499        if (argc != 3) {
500            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fspath <container-id>", false);
501            return 0;
502        }
503        char path[255];
504
505        if (!(rc = vm->getAsecFilesystemPath(argv[2], path, sizeof(path)))) {
506            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
507            return 0;
508        }
509    } else {
510        dumpArgs(argc, argv, -1);
511        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown asec cmd", false);
512    }
513
514    if (!rc) {
515        cli->sendMsg(ResponseCode::CommandOkay, "asec operation succeeded", false);
516    } else {
517        rc = ResponseCode::convertFromErrno();
518        cli->sendMsg(rc, "asec operation failed", true);
519    }
520
521    return 0;
522}
523
524CommandListener::ObbCmd::ObbCmd() :
525                 VoldCommand("obb") {
526}
527
528int CommandListener::ObbCmd::runCommand(SocketClient *cli,
529                                                      int argc, char **argv) {
530    if (argc < 2) {
531        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
532        return 0;
533    }
534
535    VolumeManager *vm = VolumeManager::Instance();
536    int rc = 0;
537
538    if (!strcmp(argv[1], "list")) {
539        dumpArgs(argc, argv, -1);
540
541        rc = vm->listMountedObbs(cli);
542    } else if (!strcmp(argv[1], "mount")) {
543            dumpArgs(argc, argv, 3);
544            if (argc != 5) {
545                cli->sendMsg(ResponseCode::CommandSyntaxError,
546                        "Usage: obb mount <filename> <key> <ownerGid>", false);
547                return 0;
548            }
549            rc = vm->mountObb(argv[2], argv[3], atoi(argv[4]));
550    } else if (!strcmp(argv[1], "unmount")) {
551        dumpArgs(argc, argv, -1);
552        if (argc < 3) {
553            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: obb unmount <source file> [force]", false);
554            return 0;
555        }
556        bool force = false;
557        if (argc > 3 && !strcmp(argv[3], "force")) {
558            force = true;
559        }
560        rc = vm->unmountObb(argv[2], force);
561    } else if (!strcmp(argv[1], "path")) {
562        dumpArgs(argc, argv, -1);
563        if (argc != 3) {
564            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: obb path <source file>", false);
565            return 0;
566        }
567        char path[255];
568
569        if (!(rc = vm->getObbMountPath(argv[2], path, sizeof(path)))) {
570            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
571            return 0;
572        }
573    } else {
574        dumpArgs(argc, argv, -1);
575        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown obb cmd", false);
576    }
577
578    if (!rc) {
579        cli->sendMsg(ResponseCode::CommandOkay, "obb operation succeeded", false);
580    } else {
581        rc = ResponseCode::convertFromErrno();
582        cli->sendMsg(rc, "obb operation failed", true);
583    }
584
585    return 0;
586}
587
588CommandListener::FstrimCmd::FstrimCmd() :
589                 VoldCommand("fstrim") {
590}
591int CommandListener::FstrimCmd::runCommand(SocketClient *cli,
592                                                      int argc, char **argv) {
593    if ((cli->getUid() != 0) && (cli->getUid() != AID_SYSTEM)) {
594        cli->sendMsg(ResponseCode::CommandNoPermission, "No permission to run fstrim commands", false);
595        return 0;
596    }
597
598    if (argc < 2) {
599        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
600        return 0;
601    }
602
603    VolumeManager *vm = VolumeManager::Instance();
604    std::lock_guard<std::mutex> lock(vm->getLock());
605
606    int flags = 0;
607
608    std::string cmd(argv[1]);
609    if (cmd == "dotrim") {
610        flags = 0;
611    } else if (cmd == "dotrimbench") {
612        flags = android::vold::TrimTask::Flags::kBenchmarkAfter;
613    } else if (cmd == "dodtrim") {
614        flags = android::vold::TrimTask::Flags::kDeepTrim;
615    } else if (cmd == "dodtrimbench") {
616        flags = android::vold::TrimTask::Flags::kDeepTrim
617                | android::vold::TrimTask::Flags::kBenchmarkAfter;
618    }
619
620    (new android::vold::TrimTask(flags))->start();
621    return sendGenericOkFail(cli, 0);
622}
623