X-Git-Url: http://git.tuebingen.mpg.de/?p=osl.git;a=blobdiff_plain;f=fsck.c;h=e814df55e866529328608d5acd39c5e2e9c76366;hp=84e624d7981a822efa923d9b1ffcfd83a2f672fd;hb=576fe981b1c21595f2b04c2b69153dc0a46dbfdf;hpb=c4d11daf607d7cfea1eaa1285dc32c17f0f0289d diff --git a/fsck.c b/fsck.c index 84e624d..e814df5 100644 --- a/fsck.c +++ b/fsck.c @@ -1,5 +1,5 @@ /* - * Copyright (C) 2007-2009 Andre Noll + * Copyright (C) 2007-2009 Andre Noll * * Licensed under the GPL v2. For licencing details see COPYING. */ @@ -9,6 +9,7 @@ #include #include #include +#include #include #include @@ -16,24 +17,15 @@ #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 .\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"), \ @@ -51,7 +43,7 @@ enum { }; #undef FSCK_ERROR #define FSCK_ERROR(num, txt) txt -const char const *fsck_errors[] = { +static const char *fsck_errors[] = { FSCK_ERRORS }; @@ -65,7 +57,7 @@ static const char *fsck_strerror(int num) extern int loglevel; /* taken from git */ -signed char hexval_table[256] = { +static signed char hexval_table[256] = { -1, -1, -1, -1, -1, -1, -1, -1, /* 00-07 */ -1, -1, -1, -1, -1, -1, -1, -1, /* 08-0f */ -1, -1, -1, -1, -1, -1, -1, -1, /* 10-17 */ @@ -100,7 +92,7 @@ signed char hexval_table[256] = { -1, -1, -1, -1, -1, -1, -1, -1, /* f8-ff */ }; -int asc_to_hash(const char *asc_hash, int len, HASH_TYPE *hash) +static int asc_to_hash(const char *asc_hash, int len, HASH_TYPE *hash) { int i = 0; const unsigned char *asc = (const unsigned char *) asc_hash; @@ -280,9 +272,10 @@ static int fsck_opendir(const char *dirname, DIR **dir, int *cwd) 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); @@ -390,7 +383,7 @@ static int check_range(struct osl_table *t, uint32_t row_num, uint32_t col_num) 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); } @@ -400,24 +393,25 @@ static int fsck_mark_row_invalid(struct osl_table *t, int 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++; @@ -474,7 +468,7 @@ static int prune_invalid_rows_from_index(struct osl_table *t) } 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; @@ -512,29 +506,30 @@ static int prune_invalid_rows_from_index(struct osl_table *t) 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; } } @@ -550,24 +545,25 @@ err: /* 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); @@ -683,7 +679,7 @@ static int check_disk_storage_column(struct osl_table *t, int row_num, (*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); @@ -709,7 +705,8 @@ out: 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; @@ -719,18 +716,18 @@ static int check_disk_storage_presence(struct osl_table *t) 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; @@ -758,7 +755,7 @@ static int dummy_compare(const struct osl_object *obj1, const struct osl_object static unsigned files_pruned; -int prune_disk_storage_file(const char *path, void *private_data) +static int prune_disk_storage_file(const char *path, void *private_data) { HASH_TYPE hash[HASH_SIZE]; uint8_t flags = *(uint8_t *)private_data; @@ -804,7 +801,7 @@ int prune_disk_storage_file(const char *path, void *private_data) 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); @@ -896,7 +893,7 @@ static int fsck_init(struct osl_table_description *desc, struct osl_table **t) } 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); @@ -930,7 +927,7 @@ static void fsck_cleanup(struct osl_table *t) #define ST_CASE(st) case st: return #st -const char *get_asc_storage_type(enum osl_storage_type st) +static const char *get_asc_storage_type(enum osl_storage_type st) { switch (st) { ST_CASE(OSL_MAPPED_STORAGE); @@ -945,7 +942,7 @@ const char *get_asc_storage_type(enum osl_storage_type st) else str = fsck_strdup(#flag); }} while (0) -char *get_asc_storage_flags(enum osl_storage_type sf) +static char *get_asc_storage_flags(enum osl_storage_type sf) { char *asc_sf = NULL; @@ -1100,7 +1097,7 @@ out: 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); @@ -1154,7 +1151,8 @@ static int fsck(struct osl_table_description *desc) 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; @@ -1167,7 +1165,7 @@ out: 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, @@ -1177,15 +1175,15 @@ static int check_table(char *db_dir, char *table_name) 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, @@ -1195,7 +1193,7 @@ out: return ret; } -static int check_all_tables(char *db_dir) +static int check_all_tables(const char *db_dir) { DIR *dir; struct dirent *entry; @@ -1229,38 +1227,60 @@ static int check_all_tables(char *db_dir) 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 .\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;