| /* |
| * Copyright (C) 2008 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <stdlib.h> |
| #include <sys/socket.h> |
| #include <sys/types.h> |
| #include <netinet/in.h> |
| #include <arpa/inet.h> |
| #include <dirent.h> |
| #include <errno.h> |
| #include <fcntl.h> |
| |
| #define LOG_TAG "VoldCmdListener" |
| #include <cutils/log.h> |
| |
| #include <sysutils/SocketClient.h> |
| |
| #include "CommandListener.h" |
| #include "VolumeManager.h" |
| #include "ResponseCode.h" |
| #include "Process.h" |
| #include "Xwarp.h" |
| #include "Loop.h" |
| #include "Devmapper.h" |
| |
| CommandListener::CommandListener() : |
| FrameworkListener("vold") { |
| registerCmd(new DumpCmd()); |
| registerCmd(new VolumeCmd()); |
| registerCmd(new AsecCmd()); |
| registerCmd(new ShareCmd()); |
| registerCmd(new StorageCmd()); |
| registerCmd(new XwarpCmd()); |
| } |
| |
| void CommandListener::dumpArgs(int argc, char **argv, int argObscure) { |
| char buffer[4096]; |
| char *p = buffer; |
| |
| memset(buffer, 0, sizeof(buffer)); |
| int i; |
| for (i = 0; i < argc; i++) { |
| int len = strlen(argv[i]) + 1; // Account for space |
| if (i == argObscure) { |
| len += 2; // Account for {} |
| } |
| if (((p - buffer) + len) < (sizeof(buffer)-1)) { |
| if (i == argObscure) { |
| *p++ = '{'; |
| *p++ = '}'; |
| *p++ = ' '; |
| continue; |
| } |
| strcpy(p, argv[i]); |
| p+= strlen(argv[i]); |
| if (i != (argc -1)) { |
| *p++ = ' '; |
| } |
| } |
| } |
| SLOGD("%s", buffer); |
| } |
| |
| CommandListener::DumpCmd::DumpCmd() : |
| VoldCommand("dump") { |
| } |
| |
| int CommandListener::DumpCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| cli->sendMsg(0, "Dumping loop status", false); |
| if (Loop::dumpState(cli)) { |
| cli->sendMsg(ResponseCode::CommandOkay, "Loop dump failed", true); |
| } |
| cli->sendMsg(0, "Dumping DM status", false); |
| if (Devmapper::dumpState(cli)) { |
| cli->sendMsg(ResponseCode::CommandOkay, "Devmapper dump failed", true); |
| } |
| cli->sendMsg(0, "Dumping mounted filesystems", false); |
| FILE *fp = fopen("/proc/mounts", "r"); |
| if (fp) { |
| char line[1024]; |
| while (fgets(line, sizeof(line), fp)) { |
| line[strlen(line)-1] = '\0'; |
| cli->sendMsg(0, line, false);; |
| } |
| fclose(fp); |
| } |
| |
| cli->sendMsg(ResponseCode::CommandOkay, "dump complete", false); |
| return 0; |
| } |
| |
| |
| CommandListener::VolumeCmd::VolumeCmd() : |
| VoldCommand("volume") { |
| } |
| |
| int CommandListener::VolumeCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| dumpArgs(argc, argv, -1); |
| |
| if (argc < 2) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false); |
| return 0; |
| } |
| |
| VolumeManager *vm = VolumeManager::Instance(); |
| int rc = 0; |
| |
| if (!strcmp(argv[1], "list")) { |
| return vm->listVolumes(cli); |
| } else if (!strcmp(argv[1], "debug")) { |
| if (argc != 3 || (argc == 3 && (strcmp(argv[2], "off") && strcmp(argv[2], "on")))) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: volume debug <off/on>", false); |
| return 0; |
| } |
| vm->setDebug(!strcmp(argv[2], "on") ? true : false); |
| } else if (!strcmp(argv[1], "mount")) { |
| if (argc != 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: volume mount <path>", false); |
| return 0; |
| } |
| rc = vm->mountVolume(argv[2]); |
| } else if (!strcmp(argv[1], "unmount")) { |
| if (argc < 3 || argc > 4 || (argc == 4 && strcmp(argv[3], "force"))) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: volume unmount <path> [force]", false); |
| return 0; |
| } |
| |
| bool force = false; |
| if (argc >= 4 && !strcmp(argv[3], "force")) { |
| force = true; |
| } |
| rc = vm->unmountVolume(argv[2], force); |
| } else if (!strcmp(argv[1], "format")) { |
| if (argc != 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: volume format <path>", false); |
| return 0; |
| } |
| rc = vm->formatVolume(argv[2]); |
| } else if (!strcmp(argv[1], "share")) { |
| if (argc != 4) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: volume share <path> <method>", false); |
| return 0; |
| } |
| rc = vm->shareVolume(argv[2], argv[3]); |
| } else if (!strcmp(argv[1], "unshare")) { |
| if (argc != 4) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: volume unshare <path> <method>", false); |
| return 0; |
| } |
| rc = vm->unshareVolume(argv[2], argv[3]); |
| } else if (!strcmp(argv[1], "shared")) { |
| bool enabled = false; |
| if (argc != 4) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: volume shared <path> <method>", false); |
| return 0; |
| } |
| |
| if (vm->shareEnabled(argv[2], argv[3], &enabled)) { |
| cli->sendMsg( |
| ResponseCode::OperationFailed, "Failed to determine share enable state", true); |
| } else { |
| cli->sendMsg(ResponseCode::ShareEnabledResult, |
| (enabled ? "Share enabled" : "Share disabled"), false); |
| } |
| return 0; |
| } else { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume cmd", false); |
| } |
| |
| if (!rc) { |
| cli->sendMsg(ResponseCode::CommandOkay, "volume operation succeeded", false); |
| } else { |
| int erno = errno; |
| rc = ResponseCode::convertFromErrno(); |
| cli->sendMsg(rc, "volume operation failed", true); |
| } |
| |
| return 0; |
| } |
| |
| CommandListener::ShareCmd::ShareCmd() : |
| VoldCommand("share") { |
| } |
| |
| int CommandListener::ShareCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| dumpArgs(argc, argv, -1); |
| |
| if (argc < 2) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false); |
| return 0; |
| } |
| |
| VolumeManager *vm = VolumeManager::Instance(); |
| int rc = 0; |
| |
| if (!strcmp(argv[1], "status")) { |
| bool avail = false; |
| |
| if (vm->shareAvailable(argv[2], &avail)) { |
| cli->sendMsg( |
| ResponseCode::OperationFailed, "Failed to determine share availability", true); |
| } else { |
| cli->sendMsg(ResponseCode::ShareStatusResult, |
| (avail ? "Share available" : "Share unavailable"), false); |
| } |
| } else { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown share cmd", false); |
| } |
| |
| return 0; |
| } |
| |
| CommandListener::StorageCmd::StorageCmd() : |
| VoldCommand("storage") { |
| } |
| |
| int CommandListener::StorageCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| dumpArgs(argc, argv, -1); |
| |
| if (argc < 2) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false); |
| return 0; |
| } |
| |
| if (!strcmp(argv[1], "users")) { |
| DIR *dir; |
| struct dirent *de; |
| |
| if (!(dir = opendir("/proc"))) { |
| cli->sendMsg(ResponseCode::OperationFailed, "Failed to open /proc", true); |
| return 0; |
| } |
| |
| while ((de = readdir(dir))) { |
| int pid = Process::getPid(de->d_name); |
| |
| if (pid < 0) { |
| continue; |
| } |
| |
| char processName[255]; |
| Process::getProcessName(pid, processName, sizeof(processName)); |
| |
| if (Process::checkFileDescriptorSymLinks(pid, argv[2]) || |
| Process::checkFileMaps(pid, argv[2]) || |
| Process::checkSymLink(pid, argv[2], "cwd") || |
| Process::checkSymLink(pid, argv[2], "root") || |
| Process::checkSymLink(pid, argv[2], "exe")) { |
| |
| char msg[1024]; |
| snprintf(msg, sizeof(msg), "%d %s", pid, processName); |
| cli->sendMsg(ResponseCode::StorageUsersListResult, msg, false); |
| } |
| } |
| closedir(dir); |
| cli->sendMsg(ResponseCode::CommandOkay, "Storage user list complete", false); |
| } else { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown storage cmd", false); |
| } |
| return 0; |
| } |
| |
| CommandListener::AsecCmd::AsecCmd() : |
| VoldCommand("asec") { |
| } |
| |
| int CommandListener::AsecCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| if (argc < 2) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false); |
| return 0; |
| } |
| |
| VolumeManager *vm = VolumeManager::Instance(); |
| int rc = 0; |
| |
| if (!strcmp(argv[1], "list")) { |
| dumpArgs(argc, argv, -1); |
| DIR *d = opendir(Volume::SEC_ASECDIR); |
| |
| if (!d) { |
| cli->sendMsg(ResponseCode::OperationFailed, "Failed to open asec dir", true); |
| return 0; |
| } |
| |
| struct dirent *dent; |
| while ((dent = readdir(d))) { |
| if (dent->d_name[0] == '.') |
| continue; |
| if (!strcmp(&dent->d_name[strlen(dent->d_name)-5], ".asec")) { |
| char id[255]; |
| memset(id, 0, sizeof(id)); |
| strncpy(id, dent->d_name, strlen(dent->d_name) -5); |
| cli->sendMsg(ResponseCode::AsecListResult, id, false); |
| } |
| } |
| closedir(d); |
| } else if (!strcmp(argv[1], "create")) { |
| dumpArgs(argc, argv, 5); |
| if (argc != 7) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: asec create <container-id> <size_mb> <fstype> <key> <ownerUid>", false); |
| return 0; |
| } |
| |
| unsigned int numSectors = (atoi(argv[3]) * (1024 * 1024)) / 512; |
| rc = vm->createAsec(argv[2], numSectors, argv[4], argv[5], atoi(argv[6])); |
| } else if (!strcmp(argv[1], "finalize")) { |
| dumpArgs(argc, argv, -1); |
| if (argc != 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec finalize <container-id>", false); |
| return 0; |
| } |
| rc = vm->finalizeAsec(argv[2]); |
| } else if (!strcmp(argv[1], "destroy")) { |
| dumpArgs(argc, argv, -1); |
| if (argc < 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec destroy <container-id> [force]", false); |
| return 0; |
| } |
| bool force = false; |
| if (argc > 3 && !strcmp(argv[3], "force")) { |
| force = true; |
| } |
| rc = vm->destroyAsec(argv[2], force); |
| } else if (!strcmp(argv[1], "mount")) { |
| dumpArgs(argc, argv, 3); |
| if (argc != 5) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: asec mount <namespace-id> <key> <ownerUid>", false); |
| return 0; |
| } |
| rc = vm->mountAsec(argv[2], argv[3], atoi(argv[4])); |
| } else if (!strcmp(argv[1], "unmount")) { |
| dumpArgs(argc, argv, -1); |
| if (argc < 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec unmount <container-id> [force]", false); |
| return 0; |
| } |
| bool force = false; |
| if (argc > 3 && !strcmp(argv[3], "force")) { |
| force = true; |
| } |
| rc = vm->unmountAsec(argv[2], force); |
| } else if (!strcmp(argv[1], "rename")) { |
| dumpArgs(argc, argv, -1); |
| if (argc != 4) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, |
| "Usage: asec rename <old_id> <new_id>", false); |
| return 0; |
| } |
| rc = vm->renameAsec(argv[2], argv[3]); |
| } else if (!strcmp(argv[1], "path")) { |
| dumpArgs(argc, argv, -1); |
| if (argc != 3) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec path <container-id>", false); |
| return 0; |
| } |
| char path[255]; |
| |
| if (!(rc = vm->getAsecMountPath(argv[2], path, sizeof(path)))) { |
| cli->sendMsg(ResponseCode::AsecPathResult, path, false); |
| return 0; |
| } |
| } else { |
| dumpArgs(argc, argv, -1); |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown asec cmd", false); |
| } |
| |
| if (!rc) { |
| cli->sendMsg(ResponseCode::CommandOkay, "asec operation succeeded", false); |
| } else { |
| rc = ResponseCode::convertFromErrno(); |
| cli->sendMsg(rc, "asec operation failed", true); |
| } |
| |
| return 0; |
| } |
| |
| CommandListener::XwarpCmd::XwarpCmd() : |
| VoldCommand("xwarp") { |
| } |
| |
| int CommandListener::XwarpCmd::runCommand(SocketClient *cli, |
| int argc, char **argv) { |
| if (argc < 2) { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false); |
| return 0; |
| } |
| |
| if (!strcmp(argv[1], "enable")) { |
| if (Xwarp::enable()) { |
| cli->sendMsg(ResponseCode::OperationFailed, "Failed to enable xwarp", true); |
| return 0; |
| } |
| |
| cli->sendMsg(ResponseCode::CommandOkay, "Xwarp mirroring started", false); |
| } else if (!strcmp(argv[1], "disable")) { |
| if (Xwarp::disable()) { |
| cli->sendMsg(ResponseCode::OperationFailed, "Failed to disable xwarp", true); |
| return 0; |
| } |
| |
| cli->sendMsg(ResponseCode::CommandOkay, "Xwarp disabled", false); |
| } else if (!strcmp(argv[1], "status")) { |
| char msg[255]; |
| bool r; |
| unsigned mirrorPos, maxSize; |
| |
| if (Xwarp::status(&r, &mirrorPos, &maxSize)) { |
| cli->sendMsg(ResponseCode::OperationFailed, "Failed to get xwarp status", true); |
| return 0; |
| } |
| snprintf(msg, sizeof(msg), "%s %u %u", (r ? "ready" : "not-ready"), mirrorPos, maxSize); |
| cli->sendMsg(ResponseCode::XwarpStatusResult, msg, false); |
| } else { |
| cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown storage cmd", false); |
| } |
| |
| return 0; |
| } |
| |
