The beginnings of the mountvolume command, using the DiskDevice API. Interrupted work. Probably doesn't even compile yet.
git-svn-id: file:///srv/svn/repos/haiku/trunk/current@11745 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Ingo Weinhold
parent
8a9c34bcfb
commit
0f6fa84153
280
src/apps/bin/mountvolume.cpp
Normal file
280
src/apps/bin/mountvolume.cpp
Normal file
@ -0,0 +1,280 @@
|
||||
// mountvolume.cpp
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include <set>
|
||||
#include <string>
|
||||
|
||||
#include <DiskDevice.h>
|
||||
#include <DiskDevicePrivate.h>
|
||||
#include <DiskDeviceRoster.h>
|
||||
#include <Partition.h>
|
||||
|
||||
typedef set<string> StringSet;
|
||||
|
||||
// usage
|
||||
static const char *kUsage =
|
||||
"%s <options> [ <volume name> ... ]\n"
|
||||
"Mounts the volume with name <volume name>, if given. Lists info about\n"
|
||||
"mounted and mountable volumes and mounts/unmounts volumes.\n"
|
||||
"\n"
|
||||
"The terminology is actually not quite correct: By volumes only partitions\n"
|
||||
"living on disk devices are meant.\n"
|
||||
"\n"
|
||||
"Options:\n"
|
||||
"[general]\n"
|
||||
" -s - silent; don't print info about (un)mounting\n"
|
||||
" -h, --help - print this info text\n"
|
||||
"\n"
|
||||
"[mounting]\n"
|
||||
" -all - mount all mountable volumes\n"
|
||||
" -allbfs - mount all mountable BFS volumes\n"
|
||||
" -allhfs - mount all mountable HFS volumes\n"
|
||||
" -alldos - mount all mountable DOS volumes\n"
|
||||
" -ro - mount volumes read-only\n"
|
||||
" -unmount <volume> - unmount the volume with the name <volume>\n"
|
||||
"\n"
|
||||
"[info]\n"
|
||||
" -p, -l - list all mounted and mountable volumes\n"
|
||||
" -lh - list all existing volumes (incl. not-mountable ones)\n"
|
||||
" -dd - list all disk existing devices\n"
|
||||
"\n"
|
||||
"[obsolete]\n"
|
||||
" -r - ignored\n"
|
||||
" -publishall - ignored\n"
|
||||
" -publishbfs - ignored\n"
|
||||
" -publishhfs - ignored\n"
|
||||
" -publishdos - ignored\n"
|
||||
;
|
||||
|
||||
// application name
|
||||
const char *kAppName = "mountvolume";
|
||||
|
||||
// print_usage
|
||||
static
|
||||
void
|
||||
print_usage(const char *appName, bool error)
|
||||
{
|
||||
if (appName) {
|
||||
if (const char *lastSlash = strrchr(appName, '/')
|
||||
appName = lastSlash
|
||||
} else
|
||||
appName = kAppName;
|
||||
fprintf((error ? stderr : stdout), kUsage, appName);
|
||||
}
|
||||
|
||||
// print_usage_and_exit
|
||||
static
|
||||
print_usage_and_exit(const char *appName, bool error)
|
||||
{
|
||||
print_usage(appName, error);
|
||||
exit(error ? 0 : 1);
|
||||
}
|
||||
|
||||
// MountVisitor
|
||||
struct MountVisitor : public BDiskDeviceVisitor {
|
||||
MountVisitor()
|
||||
: silent(false),
|
||||
mountAll(false),
|
||||
mountBFS(false),
|
||||
mountHFS(false),
|
||||
mountDOS(false),
|
||||
readOnly(false)
|
||||
{
|
||||
}
|
||||
|
||||
virtual bool Visit(BDiskDevice *device)
|
||||
{
|
||||
Visit(device, 0);
|
||||
}
|
||||
|
||||
virtual bool Visit(BPartition *partition, int32 level)
|
||||
{
|
||||
// get name and type
|
||||
const char *name = partition->ContentName();
|
||||
if (!name)
|
||||
name = partition->Name();
|
||||
const char *type = partition->ContentType();
|
||||
|
||||
// check whether to mount
|
||||
bool mount = false;
|
||||
if (name && toMount.find(name) != toMount.end()) {
|
||||
toMount.erase(name);
|
||||
if (!partition->IsMounted()
|
||||
mount = true;
|
||||
else if (!silent)
|
||||
fprintf(stderr, "Volume `%s' already mounted.\n", name);
|
||||
} else if (mountAll) {
|
||||
mount = true;
|
||||
} else if (mountBFS && contentType
|
||||
&& strcmp(contentType, kPartitionTypeBFS) == 0) {
|
||||
mount = true;
|
||||
} else if (mountHFS && contentType
|
||||
&& strcmp(contentType, kPartitionTypeHFS) == 0) {
|
||||
mount = true;
|
||||
} else if (mountDOS && contentType
|
||||
&& (strcmp(contentType, kPartitionTypeFAT12) == 0
|
||||
|| strcmp(contentType, kPartitionTypeFAT32) == 0)) {
|
||||
mount = true;
|
||||
}
|
||||
|
||||
// check whether to unmount
|
||||
bool unmount = false;
|
||||
if (name && toUnmount.find(name) != toUnmount.end()) {
|
||||
toUnmount.erase(name);
|
||||
if (!partition->IsMounted() {
|
||||
unmount = true;
|
||||
mount = false;
|
||||
} else if (!silent)
|
||||
fprintf(stderr, "Volume `%s' not mounted.\n", name);
|
||||
}
|
||||
|
||||
// mount/unmount
|
||||
if (mount) {
|
||||
status_t error = partition->Mount(NULL,
|
||||
(readOnly ? B_MOUNT_READ_ONLY : 0));
|
||||
if (!silent) {
|
||||
if (error == B_OK) {
|
||||
printf("Volume `%s' mounted successfully.\n", name);
|
||||
} else {
|
||||
fprintf("Failed to mount volume `%s': %s\n", name,
|
||||
strerror(error));
|
||||
}
|
||||
}
|
||||
} else if (unmount) {
|
||||
status_t error = partition->Unmount();
|
||||
if (!silent) {
|
||||
if (error == B_OK) {
|
||||
printf("Volume `%s' unmounted successfully.\n", name);
|
||||
} else {
|
||||
fprintf("Failed to unmount volume `%s': %s\n", name,
|
||||
strerror(error));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool silent;
|
||||
StringSet toMount;
|
||||
StringSet toUnmount;
|
||||
bool mountAll;
|
||||
bool mountBFS;
|
||||
bool mountHFS;
|
||||
bool mountDOS;
|
||||
bool readOnly;
|
||||
};
|
||||
|
||||
// PrintPartitionsVisitor
|
||||
struct PrintPartitionsVisitor : public BDiskDeviceVisitor {
|
||||
PrintPartitionsVisitor()
|
||||
: listMountablePartitions(false),
|
||||
listAllPartitions(false)
|
||||
{
|
||||
}
|
||||
|
||||
virtual bool Visit(BDiskDevice *device)
|
||||
{
|
||||
Visit(device, 0);
|
||||
}
|
||||
|
||||
virtual bool Visit(BPartition *partition, int32 level)
|
||||
{
|
||||
}
|
||||
|
||||
bool listMountablePartitions;
|
||||
bool listAllPartitions;
|
||||
};
|
||||
|
||||
|
||||
// main
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
if (argc < 2)
|
||||
print_usage_and_exit(argv[0], true);
|
||||
|
||||
MountVisitor mountVisitor;
|
||||
PrintPartitionsVisitor printPartitionsVisitor;
|
||||
bool listAllDevices = false;
|
||||
|
||||
// parse arguments
|
||||
int argi = 1;
|
||||
while (argi < argc) {
|
||||
const char *arg = argv[argi];
|
||||
|
||||
if (arg[0] != '\0' && arg[0] != '-') {
|
||||
mountVisitor.toMount.insert(arg);
|
||||
} else if (strcmp(arg, "-s") == 0) {
|
||||
mountVisitor.silent = true;
|
||||
} else if (strcmp(arg, "-h") == 0 || strcmp(arg, "--help") == 0) {
|
||||
print_usage_and_exit(argv[0], false);
|
||||
} else if (strcmp(arg, "-all") == 0) {
|
||||
mountVisitor.mountAll = true;
|
||||
} else if (strcmp(arg, "-allbfs") == 0) {
|
||||
mountVisitor.mountAllBFS = true;
|
||||
} else if (strcmp(arg, "-allhfs") == 0) {
|
||||
mountVisitor.mountAllHFS = true;
|
||||
} else if (strcmp(arg, "-alldos") == 0) {
|
||||
mountVisitor.mountAllDOS = true;
|
||||
} else if (strcmp(arg, "-ro") == 0) {
|
||||
mountVisitor.readOnly = true;
|
||||
} else if (strcmp(arg, "-unmount") == 0) {
|
||||
argi++;
|
||||
if (argi >= argc)
|
||||
print_usage_and_exit(argv[0], true);
|
||||
mountVisitor.toUnmount.insert(argv[argi]);
|
||||
} else if (strcmp(arg, "-p") == 0 || strcmp(arg, "-l") == 0) {
|
||||
printPartitionsVisitor.listMountablePartitions = true;
|
||||
} else if (strcmp(arg, "-lh") == 0) {
|
||||
printPartitionsVisitor.listAllPartitions = true;
|
||||
} else if (strcmp(arg, "-dd") == 0) {
|
||||
listAllDevices = true;
|
||||
} else if (strcmp(arg, "-r") == 0 || strcmp(arg, "-publishall") == 0
|
||||
|| strcmp(arg, "-publishbfs") == 0
|
||||
|| strcmp(arg, "-publishhfs") == 0
|
||||
|| strcmp(arg, "-publishdos") == 0) {
|
||||
// obsolete: ignore
|
||||
} else
|
||||
print_usage_and_exit(argv[0], true);
|
||||
}
|
||||
|
||||
// get a disk device list
|
||||
BDiskDeviceList deviceList;
|
||||
status_t error = deviceList.Fetch();
|
||||
if (error != B_OK) {
|
||||
fprintf(stderr, "Failed to get the list of disk devices: %s",
|
||||
strerror(error));
|
||||
exit(1);
|
||||
}
|
||||
|
||||
// mount/unmount volumes
|
||||
deviceList.VisitEachMountablePartition(&mountVisitor);
|
||||
|
||||
// print errors for the volumes to mount/unmount, that weren't found
|
||||
if (!mountVisitor.silent) {
|
||||
for (StringSet::iterator it = mountVisitor.toMount.begin();
|
||||
it != mountVisitor.toMount.end();
|
||||
it++) {
|
||||
fprintf("Failed to mount volume `%s': Volume not found.\n", *it);
|
||||
}
|
||||
for (StringSet::iterator it = mountVisitor.toUnmount.begin();
|
||||
it != mountVisitor.toUnmount.end();
|
||||
it++) {
|
||||
fprintf("Failed to unmount volume `%s': Volume not found.\n", *it);
|
||||
}
|
||||
}
|
||||
|
||||
// update the disk device list
|
||||
error = deviceList.Fetch();
|
||||
if (error != B_OK) {
|
||||
fprintf(stderr, "Failed to update the list of disk devices: %s",
|
||||
strerror(error));
|
||||
exit(1);
|
||||
}
|
||||
|
||||
// print information
|
||||
// if (//
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user