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
CommandListener()53 CommandListener::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
dumpArgs(int argc,char ** argv,int argObscure)64 void 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
dumpArgs(int,char **,int)92 void CommandListener::dumpArgs(int /*argc*/, char ** /*argv*/, int /*argObscure*/) { }
93 #endif
94
sendGenericOkFail(SocketClient * cli,int cond)95 int 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
DumpCmd()103 CommandListener::DumpCmd::DumpCmd() :
104 VoldCommand("dump") {
105 }
106
runCommand(SocketClient * cli,int,char **)107 int 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
VolumeCmd()132 CommandListener::VolumeCmd::VolumeCmd() :
133 VoldCommand("volume") {
134 }
135
runCommand(SocketClient * cli,int argc,char ** argv)136 int 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
StorageCmd()277 CommandListener::StorageCmd::StorageCmd() :
278 VoldCommand("storage") {
279 }
280
runCommand(SocketClient * cli,int argc,char ** argv)281 int 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
AsecCmd()344 CommandListener::AsecCmd::AsecCmd() :
345 VoldCommand("asec") {
346 }
347
listAsecsInDirectory(SocketClient * cli,const char * directory)348 void 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
runCommand(SocketClient * cli,int argc,char ** argv)386 int 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
ObbCmd()524 CommandListener::ObbCmd::ObbCmd() :
525 VoldCommand("obb") {
526 }
527
runCommand(SocketClient * cli,int argc,char ** argv)528 int 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
FstrimCmd()588 CommandListener::FstrimCmd::FstrimCmd() :
589 VoldCommand("fstrim") {
590 }
runCommand(SocketClient * cli,int argc,char ** argv)591 int 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