wipefs: use symbolic value for markup mode

In case mode value is found to be out of bounds program will crash.

Signed-off-by: Sami Kerola <kerolasa@iki.fi>
This commit is contained in:
Sami Kerola 2012-06-10 22:20:34 +02:00
parent f31d041aaf
commit 4f0537374f
1 changed files with 8 additions and 3 deletions

View File

@ -56,8 +56,10 @@ struct wipe_desc {
struct wipe_desc *next; struct wipe_desc *next;
}; };
#define WP_MODE_PRETTY 0 /* default */ enum {
#define WP_MODE_PARSABLE 1 WP_MODE_PRETTY, /* default */
WP_MODE_PARSABLE
};
static const char *type_pattern; static const char *type_pattern;
@ -117,6 +119,8 @@ print_all(struct wipe_desc *wp, int mode)
case WP_MODE_PARSABLE: case WP_MODE_PARSABLE:
print_parsable(wp, n++); print_parsable(wp, n++);
break; break;
default:
abort();
} }
wp = wp->next; wp = wp->next;
} }
@ -375,7 +379,8 @@ int
main(int argc, char **argv) main(int argc, char **argv)
{ {
struct wipe_desc *wp0 = NULL, *wp; struct wipe_desc *wp0 = NULL, *wp;
int c, all = 0, has_offset = 0, noact = 0, mode = 0, quiet = 0; int c, all = 0, has_offset = 0, noact = 0, quiet = 0;
int mode = WP_MODE_PRETTY;
static const struct option longopts[] = { static const struct option longopts[] = {
{ "all", 0, 0, 'a' }, { "all", 0, 0, 'a' },