Added support for pattern matching of attribute/index names (via glob()).

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@18042 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Axel Dörfler
2006-07-05 14:22:48 +00:00
parent 38f9b78d6c
commit 79b25dd765
2 changed files with 267 additions and 149 deletions
+150 -113
View File
@@ -14,11 +14,17 @@
#include <errno.h>
#include <getopt.h>
#include <glob.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
extern const char *__progname;
const char *kProgramName = __progname;
dev_t gCurrentDevice;
// The following enum and #define are copied from gnu/sys2.h, because it
// didn't want to compile when including that directly. Since that file
// is marked as being temporary and getting migrated into system.h,
@@ -27,125 +33,26 @@
/* These enum values cannot possibly conflict with the option values
ordinarily used by commands, including CHAR_MAX + 1, etc. Avoid
CHAR_MIN - 1, as it may equal -1, the getopt end-of-options value. */
enum
{
enum {
GETOPT_HELP_CHAR = (CHAR_MIN - 2),
GETOPT_VERSION_CHAR = (CHAR_MIN - 3)
};
#define GETOPT_HELP_OPTION_DECL \
"help", no_argument, 0, GETOPT_HELP_CHAR
static struct option const longopts[] = {
{"volume", required_argument, 0, 'd'},
{"type", required_argument, 0, 't'},
{"pattern", no_argument, 0, 'p'},
{"verbose", no_argument, 0, 'v'},
{GETOPT_HELP_OPTION_DECL},
{"help", no_argument, 0, GETOPT_HELP_CHAR},
{0, 0, 0, 0}
};
void usage(int);
const char* lookup_index_type(uint32);
int
main (int32 argc, char **argv)
{
int c;
int verbose = 0;
dev_t vol_device = 0;
char *index_name = NULL;
int retval;
char *error_message;
char *path = NULL;
while ((c = getopt_long(argc, argv, "d:ht:v", longopts, NULL)) != -1) {
switch (c) {
case 0:
break;
case 'd':
path = optarg;
break;
case GETOPT_HELP_CHAR:
usage(0);
break;
case 'v':
verbose = 1;
break;
default:
usage(1);
break;
}
}
/* Remove the index from the volume of the current
directory if no volume was specified. */
if (path == NULL) {
path = ".";
}
vol_device = dev_for_path(path);
if (vol_device < 0) {
fprintf(stderr, "%s: can't get information about current volume\n", argv[0]);
return B_ERROR;
}
if (argc - optind == 1) { // last argument
index_name = argv[optind];
} else {
usage(1);
}
if (verbose) {
/* Get the index type. */
index_info i_info;
status_t status = fs_stat_index(vol_device, index_name, &i_info);
if (status != B_OK) {
fprintf(stderr, "%s: fs_stat_index(): (%d) %s\n", argv[0], errno, strerror(errno));
return (errno);
}
/* Look up the string name equivalent of the index type. */
const char* index_type_str = lookup_index_type(i_info.type);
fprintf(stdout, "Removing index \"%s\" of type %s from volume containing %s\n", index_name, index_type_str, path);
}
if ((retval = fs_remove_index(vol_device, index_name)) != 0) {
switch (errno) {
case B_FILE_ERROR:
error_message = "A filesystem error prevented the operation.";
break;
case B_BAD_VALUE:
error_message = "Invalid device specified.";
break;
case B_NOT_ALLOWED:
error_message = "Can't remove a system-reserved index, or the device is read-only.";
break;
case B_NO_MEMORY:
error_message = "Insufficient memory to complete the operation.";
break;
case B_ENTRY_NOT_FOUND:
error_message = "The specified index does not exist.";
break;
default:
error_message = "An unknown error has occured.";
break;
}
fprintf(stderr, "%s\n", error_message);
}
return 0;
}
void
usage (int status)
void
usage(int status)
{
fprintf (stderr,
"Usage: rmindex [OPTION]... INDEX_NAME\n"
"Usage: %s [OPTION]... INDEX_NAME\n"
"\n"
"Removes the index named INDEX_NAME from a disk volume. Once this has been\n"
"done, it will no longer be possible to use the query system to search for\n"
@@ -154,11 +61,13 @@ usage (int status)
" -d, --volume=PATH a path on the volume from which the index will be\n"
" removed\n"
" -h, --help display this help and exit\n"
" -v, --verbose print information about the index being removed\n"
" -p, --pattern INDEX_NAME is a pattern\n"
" -v, --verbose print information about the index being removed\n"
"\n"
"INDEX_NAME is the name of a file attribute.\n"
"\n"
"If no volume is specified, the volume of the current directory is assumed.\n");
"If no volume is specified, the volume of the current directory is assumed.\n",
kProgramName);
exit(status);
}
@@ -170,21 +79,149 @@ lookup_index_type(uint32 device_type)
switch (device_type) {
case B_DOUBLE_TYPE:
return "double";
break;
case B_FLOAT_TYPE:
return "float";
break;
case B_INT64_TYPE:
return "llong";
break;
case B_INT32_TYPE:
return "int";
break;
case B_STRING_TYPE:
return "string";
break;
default:
return "unknown";
break;
}
}
int
remove_index(dev_t device, const char* indexName, bool verbose)
{
if (verbose) {
// Get the index type
index_info info;
status_t status = fs_stat_index(device, indexName, &info);
if (status != B_OK) {
fprintf(stderr, "%s: Can't get type of index %s: %s\n",
kProgramName, indexName, strerror(errno));
return -1;
}
fprintf(stdout, "Removing index \"%s\" of type %s.\n",
indexName, lookup_index_type(info.type));
}
if (fs_remove_index(device, indexName) != 0) {
fprintf(stderr, "%s: Cannot remove index %s: %s\n", kProgramName, indexName, strerror(errno));
return -1;
}
return 0;
}
void *
open_index_dir(const char* /*path*/)
{
return fs_open_index_dir(gCurrentDevice);
}
int
stat_index(const char* /*index*/, struct stat* stat)
{
memset(stat, 0, sizeof(struct stat));
stat->st_mode = S_IFREG;
return 0;
}
int
remove_indices(dev_t device, const char* indexPattern, bool verbose)
{
glob_t glob;
memset(&glob, 0, sizeof(glob_t));
glob.gl_closedir = (void (*)(void *))fs_close_index_dir;
glob.gl_readdir = fs_read_index_dir;
glob.gl_opendir = open_index_dir;
glob.gl_lstat = stat_index;
glob.gl_stat = stat_index;
// for open_attr_dir():
gCurrentDevice = device;
int result = ::glob(indexPattern, GLOB_ALTDIRFUNC, NULL, &glob);
if (result < 0) {
errno = B_BAD_VALUE;
return -1;
}
bool error = false;
for (int i = 0; i < glob.gl_pathc; i++) {
if (remove_index(device, glob.gl_pathv[i], verbose) != 0)
error = true;
}
return error ? -1 : 0;
}
int
main(int32 argc, char **argv)
{
bool isPattern = false;
bool verbose = false;
dev_t device = 0;
char *indexName = NULL;
char *path = NULL;
int c;
while ((c = getopt_long(argc, argv, "d:ht:pv", longopts, NULL)) != -1) {
switch (c) {
case 0:
break;
case 'd':
path = optarg;
break;
case GETOPT_HELP_CHAR:
usage(0);
break;
case 'p':
isPattern = true;
break;
case 'v':
verbose = true;
break;
default:
usage(1);
break;
}
}
// Remove the index from the volume of the current
// directory if no volume was specified.
if (path == NULL)
path = ".";
device = dev_for_path(path);
if (device < 0) {
fprintf(stderr, "%s: can't get information about volume %s\n", kProgramName, path);
return 1;
}
if (argc - optind == 1) {
// last argument
indexName = argv[optind];
} else
usage(1);
int result;
if (isPattern)
result = remove_indices(device, indexName, verbose);
else
result = remove_index(device, indexName, verbose);
return result == 0 ? 0 : 1;
}