/*
- * Copyright (C) 2007-2009 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2007-2009 Andre Noll <maan@tuebingen.mpg.de>
*
* Licensed under the GPL v2. For licencing details see COPYING.
*/
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
+#include <lopsub.h>
#include <assert.h>
#include <pwd.h>
#include "osl.h"
#include "util.h"
#include "osl_core.h"
-#include "fsck.cmdline.h"
-
-/** version text used by various commands if -V switch was given */
-#define VERSION_TEXT(prefix) "osl_" prefix " " VERSION " " "\n" \
- "Copyright (C) 2008-2009 Andre Noll\n" \
- "This is free software with ABSOLUTELY NO WARRANTY." \
- " See COPYING for details.\n" \
- "Written by Andre Noll.\n" \
- "Report bugs to <maan@systemlinux.org>.\n"
-
-/** print out \p VERSION_TEXT and exit if version flag was given */
-#define HANDLE_VERSION_FLAG(_prefix, _args_info_struct) \
- if (_args_info_struct.version_given) { \
- printf("%s", VERSION_TEXT(_prefix)); \
- exit(EXIT_SUCCESS); \
- }
+#include "oslfsck.lsg.h"
-static struct fsck_args_info conf;
+static struct lls_parse_result *lpr;
+#define CMD_PTR (lls_cmd(0, oslfsck_suite))
+#define OPT_RESULT(_name) \
+ (lls_opt_result(LSG_OSLFSCK_OSLFSCK_OPT_ ## _name, lpr))
+#define OPT_GIVEN(_name) (lls_opt_given(OPT_RESULT(_name)))
+#define OPT_STRING_VAL(_name) (lls_string_val(0, OPT_RESULT(_name)))
+#define OPT_UINT32_VAL(_name) (lls_uint32_val(0, OPT_RESULT(_name)))
#define FSCK_ERRORS \
FSCK_ERROR(RANGE_VIOLATION, "range violation detected, very bad"), \
if (*dir)
return 1;
ret = errno == EACCES? -E_FSCK_ACCESS : -E_FSCK_OPENDIR;
-/* Ignore return value of fchdir() and close(). We're busted anyway. */
- if (cwd)
- fchdir(*cwd);
+ /* Ignore return value of fchdir() and close(). We're busted anyway. */
+ if (cwd) {
+ int __a_unused ret2 = fchdir(*cwd); /* STFU, gcc */
+ }
close_cwd:
if (cwd)
close(*cwd);
static int fsck_mark_row_invalid(struct osl_table *t, int i)
{
- if (conf.dry_run_given)
+ if (OPT_GIVEN(DRY_RUN))
return 0;
return mark_row_invalid(t, i);
}
*/
static int check_index_ranges(struct osl_table *t)
{
- int i, j, ret;
+ int ret;
+ unsigned k, n;
INFO_LOG("checking for range violations in index\n");
//DEBUG_LOG("%d rows. %d columns\n", t->num_rows, t->desc->num_columns);
t->num_invalid_rows = 0;
- for (i = 0; i < t->num_rows; i++) {
+ for (n = 0; n < t->num_rows; n++) {
const struct osl_column_description *cd;
- if (row_is_invalid(t, i)) {
+ if (row_is_invalid(t, n)) {
t->num_invalid_rows++;
continue;
}
- FOR_EACH_MAPPED_COLUMN(j, t, cd) {
- ret = check_range(t, i, j);
+ FOR_EACH_MAPPED_COLUMN(k, t, cd) {
+ ret = check_range(t, n, k);
if (ret < 0) {
if (ret != -E_FSCK_RANGE_VIOLATION)
goto err;
- ret = fsck_mark_row_invalid(t, i);
+ ret = fsck_mark_row_invalid(t, n);
if (ret < 0)
goto err;
t->num_invalid_rows++;
}
NOTICE_LOG("index contains %u invalid row(s) (%d bytes)\n",
t->num_invalid_rows, t->row_index_size * t->num_invalid_rows);
- if (conf.dry_run_given)
+ if (OPT_GIVEN(DRY_RUN))
return 0;
NOTICE_LOG("removing invalid rows from index\n");
bottom = t->num_rows - 1;
static int check_for_invalid_objects(struct osl_table *t, uint32_t **lost_bytes)
{
- int i, j, ret;
+ int ret;
+ unsigned k, n;
const struct osl_column_description *cd;
uint32_t *loss = fsck_malloc(sizeof(uint32_t) * t->desc->num_columns);
INFO_LOG("looking for mapped objects not contained in index\n");
/* first count used bytes */
- FOR_EACH_MAPPED_COLUMN(i, t, cd) {
- loss[i] = t->columns[i].data_map.size;
- DEBUG_LOG("column %i data map: %zu bytes\n", i,
- t->columns[i].data_map.size);
- for (j = 0; j < t->num_rows; j++) {
+ FOR_EACH_MAPPED_COLUMN(k, t, cd) {
+ loss[k] = t->columns[k].data_map.size;
+ DEBUG_LOG("column %i data map: %zu bytes\n", k,
+ t->columns[k].data_map.size);
+ for (n = 0; n < t->num_rows; n++) {
struct osl_object obj;
- ret = get_mapped_object(t, i, j, &obj);
+ ret = get_mapped_object(t, k, n, &obj);
if (ret < 0)
goto err;
- loss[i] -= obj.size;
+ loss[k] -= obj.size;
}
}
ret = 0;
- FOR_EACH_MAPPED_COLUMN(i, t, cd) {
- if (loss[i]) {
+ FOR_EACH_MAPPED_COLUMN(k, t, cd) {
+ if (loss[k]) {
NOTICE_LOG("column %u contains %u lost bytes\n",
- i, loss[i]);
+ k, loss[k]);
ret = 1;
}
}
/* prune_invalid_rows() must be run on the table before calling this */
static int prune_mapped_column(struct osl_table *t, uint32_t col_num, int fd)
{
- int i, ret;
+ int ret;
+ unsigned n;
uint32_t written = 0;
struct osl_column *col = t->columns + col_num;
INFO_LOG("pruning col %u\n", col_num);
- for (i = 0; i < t->num_rows; i++) {
+ for (n = 0; n < t->num_rows; n++) {
struct osl_object obj;
char *index_entry;
- DEBUG_LOG("checking row %u/%u\n", i, t->num_rows);
- ret = get_mapped_object(t, col_num, i, &obj);
+ DEBUG_LOG("checking row %u/%u\n", n, t->num_rows);
+ ret = get_mapped_object(t, col_num, n, &obj);
if (ret < 0)
return ret;
ret = _write_all(fd, (char *)(obj.data), obj.size);
if (ret < 0)
return ret;
written += obj.size;
- ret = get_row_index(t, i, &index_entry);
+ ret = get_row_index(t, n, &index_entry);
if (ret < 0)
return ret;
update_cell_index(index_entry, col, written, obj.size);
(*num_missing_objects)++;
ERROR_LOG("row %d: object %s is missing\n", row_num, path);
ret = 0;
- if (conf.dry_run_given)
+ if (OPT_GIVEN(DRY_RUN))
goto out;
NOTICE_LOG("trying to delete row %d\n", row_num);
ret = osl_get_row(t, dsnc, &obj, &row);
static int check_disk_storage_presence(struct osl_table *t)
{
- int ret, i, j;
+ int ret;
+ unsigned k, n;
struct osl_object obj, hash_obj = {.size = HASH_SIZE};
char *ds_name;
const struct osl_column_description *cd;
return 1;
hashes = fsck_malloc(t->num_rows * HASH_SIZE);
INFO_LOG("looking for missing disk storage objects\n");
- for (i = 0; i < t->num_rows; i++) {
- if (row_is_invalid(t, i))
+ for (k = 0; k < t->num_rows; k++) {
+ if (row_is_invalid(t, k))
continue;
- ret = get_mapped_object(t, dsnc, i, &obj);
+ ret = get_mapped_object(t, dsnc, k, &obj);
if (ret < 0)
return ret;
- hash_object(&obj, hashes + i * HASH_SIZE);
- hash_obj.data = hashes + i * HASH_SIZE;
+ hash_object(&obj, hashes + k * HASH_SIZE);
+ hash_obj.data = hashes + k * HASH_SIZE;
osl_add_row(hash_tree_table, &hash_obj);
- ds_name = disk_storage_name_of_hash(t, hashes + i * HASH_SIZE);
- FOR_EACH_DISK_STORAGE_COLUMN(j, t, cd) {
- ret = check_disk_storage_column(t, i, j, ds_name,
+ ds_name = disk_storage_name_of_hash(t, hashes + k * HASH_SIZE);
+ FOR_EACH_DISK_STORAGE_COLUMN(n, t, cd) {
+ ret = check_disk_storage_column(t, k, n, ds_name,
&missing_objects);
if (ret < 0)
goto err;
invalid:
ERROR_LOG("could not read hash value of %s\n", path);
remove:
- if (conf.dry_run_given)
+ if (OPT_GIVEN(DRY_RUN))
return 0;
NOTICE_LOG("removing %s\n", path);
unlink(path);
}
DEBUG_LOG("unmapping index\n");
osl_munmap(map.data, map.size);
- if (conf.force_given)
+ if (OPT_GIVEN(FORCE))
ret = map_table(*t, (MAP_TBL_FL_IGNORE_DIRTY));
else
ret = map_table(*t, 0);
return ret;
}
-static int dump_table(char *dump_dir, struct osl_table_description *desc)
+static int dump_table(const char *dump_dir, struct osl_table_description *desc)
{
struct osl_table *t = NULL;
int fd, ret = fsck_init(desc, &t);
ret = check_for_invalid_objects(t, &lost_bytes);
if (ret < 0)
goto out_unmap;
- if (ret > 0 && !conf.dry_run_given) { /* at least one mapped data file needs pruning */
+ if (ret > 0 && !OPT_GIVEN(DRY_RUN)) {
+ /* at least one mapped data file needs pruning */
ret = prune_objects(t, lost_bytes);
if (ret < 0)
goto out_unmap;
return ret;
}
-static int check_table(char *db_dir, char *table_name)
+static int check_table(const char *db_dir, const char *table_name)
{
struct osl_table_description desc = {
.column_descriptions = NULL,
int ret;
INFO_LOG("checking table %s\n", table_name);
- if (!conf.no_fsck_given) {
+ if (!OPT_GIVEN(NO_FSCK)) {
ret = fsck(&desc);
if (ret < 0)
goto out;
}
ret = 1;
- if (!conf.dump_dir_given || !*conf.dump_dir_arg)
+ if (!OPT_GIVEN(DUMP_DIR) || !*OPT_STRING_VAL(DUMP_DIR))
goto out;
- ret = dump_table(conf.dump_dir_arg, &desc);
+ ret = dump_table(OPT_STRING_VAL(DUMP_DIR), &desc);
out:
if (ret < 0)
ERROR_LOG("failed to check table %s: %s\n", table_name,
return ret;
}
-static int check_all_tables(char *db_dir)
+static int check_all_tables(const char *db_dir)
{
DIR *dir;
struct dirent *entry;
int main(int argc, char **argv)
{
- int i, ret;
- struct fsck_cmdline_parser_params params = {
- .override = 0,
- .initialize = 1,
- .check_required = 1,
- .check_ambiguity = 1,
- .print_errors = 1
- };
- ret = fsck_cmdline_parser_ext(argc, argv, &conf, ¶ms);
+ int ret;
+ unsigned n;
+ char *errctx = NULL;
+ const char *dd;
+
+ loglevel = 0;
+ ret = lls_parse(argc, argv, CMD_PTR, &lpr, &errctx);
if (ret < 0) {
- loglevel = EMERG;
- ret = -E_FSCK_SYNTAX;
- goto out;
+ if (errctx)
+ ERROR_LOG("%s\n", errctx);
+ EMERG_LOG("%s\n", lls_strerror(-ret));
+ exit(EXIT_FAILURE);
+ }
+ loglevel = OPT_UINT32_VAL(LOGLEVEL);
+ if (OPT_GIVEN(DETAILED_HELP)) {
+ printf("%s\n", lls_long_help(CMD_PTR));
+ exit(EXIT_SUCCESS);
+ }
+ if (OPT_GIVEN(HELP)) {
+ printf("%s\n", lls_short_help(CMD_PTR));
+ exit(EXIT_SUCCESS);
+ }
+ if (OPT_GIVEN(VERSION)) {
+ printf(
+ "oslfsck " OSL_VERSION "\n"
+ "Copyright (C) 2008-2009 Andre Noll\n"
+ "This is free software with ABSOLUTELY NO WARRANTY."
+ " See COPYING for details.\n"
+ "Written by Andre Noll.\n"
+ "Report bugs to <maan@tuebingen.mpg.de>.\n"
+ );
+ exit(EXIT_SUCCESS);
+ }
+ dd = OPT_STRING_VAL(DATABASE_DIR);
+ if (!dd) {
+ EMERG_LOG("--database-dir: mandatory option not given\n");
+ exit(EXIT_FAILURE);
}
- loglevel = conf.loglevel_arg;
- HANDLE_VERSION_FLAG("fsck", conf);
- INFO_LOG("database dir: %s\n", conf.database_dir_arg);
- if (!conf.inputs_num) {
- ret = check_all_tables(conf.database_dir_arg);
+ INFO_LOG("database dir: %s\n", dd);
+ if (lls_num_inputs(lpr) == 0) {
+ ret = check_all_tables(dd);
goto out;
}
- for (i = 0; i < conf.inputs_num; i++) {
- ret = check_table(conf.database_dir_arg, conf.inputs[i]);
+ for (n = 0; n < lls_num_inputs(lpr); n++) {
+ ret = check_table(dd, lls_input(n, lpr));
if (ret < 0)
break;
}
out:
if (ret < 0) {
ERROR_LOG("%s\n", fsck_strerror(-ret));
- if (conf.loglevel_arg > 1)
+ if (loglevel > 1)
EMERG_LOG("re-run with \"--loglevel %d\" to increase verbosity\n",
- conf.loglevel_arg - 1);
+ loglevel - 1);
} else
NOTICE_LOG("success\n");
return ret < 0? EXIT_FAILURE : EXIT_SUCCESS;
projects / osl.git / blobdiff