static void
check_status(int fd, int meter)
{
int recon_percent_done = 0;
int parity_percent_done = 0;
int copyback_percent_done = 0;
do_ioctl(fd, RAIDFRAME_CHECK_RECON_STATUS, &recon_percent_done,
"RAIDFRAME_CHECK_RECON_STATUS");
printf("Reconstruction is %d%% complete.\n", recon_percent_done);
do_ioctl(fd, RAIDFRAME_CHECK_PARITYREWRITE_STATUS,
&parity_percent_done,
"RAIDFRAME_CHECK_PARITYREWRITE_STATUS");
printf("Parity Re-write is %d%% complete.\n", parity_percent_done);
do_ioctl(fd, RAIDFRAME_CHECK_COPYBACK_STATUS, ©back_percent_done,
"RAIDFRAME_CHECK_COPYBACK_STATUS");
printf("Copyback is %d%% complete.\n", copyback_percent_done);
if (meter) {
/* These 3 should be mutually exclusive at this point */
if (recon_percent_done < 100) {
printf("Reconstruction status:\n");
do_meter(fd,RAIDFRAME_CHECK_RECON_STATUS_EXT);
} else if (parity_percent_done < 100) {
printf("Parity Re-write status:\n");
do_meter(fd,RAIDFRAME_CHECK_PARITYREWRITE_STATUS_EXT);
} else if (copyback_percent_done < 100) {
printf("Copyback status:\n");
do_meter(fd,RAIDFRAME_CHECK_COPYBACK_STATUS_EXT);
}
}
}
static void
check_parity(int fd, int do_rewrite, char *dev_name)
{
int is_clean;
int percent_done;
is_clean = 0;
percent_done = 0;
do_ioctl(fd, RAIDFRAME_CHECK_PARITY, &is_clean,
"RAIDFRAME_CHECK_PARITY");
if (is_clean) {
printf("%s: Parity status: clean\n",dev_name);
} else {
printf("%s: Parity status: DIRTY\n",dev_name);
if (do_rewrite) {
printf("%s: Initiating re-write of parity\n",
dev_name);
do_ioctl(fd, RAIDFRAME_REWRITEPARITY, NULL,
"RAIDFRAME_REWRITEPARITY");
sleep(3); /* XXX give it time to
get started. */
if (verbose) {
printf("Parity Re-write status:\n");
do_meter(fd, RAIDFRAME_CHECK_PARITYREWRITE_STATUS_EXT);
} else {
do_ioctl(fd,
RAIDFRAME_CHECK_PARITYREWRITE_STATUS,
&percent_done,
"RAIDFRAME_CHECK_PARITYREWRITE_STATUS"
);
while( percent_done < 100 ) {
sleep(3); /* wait a bit... */
do_ioctl(fd, RAIDFRAME_CHECK_PARITYREWRITE_STATUS,
&percent_done, "RAIDFRAME_CHECK_PARITYREWRITE_STATUS");
}
}
printf("%s: Parity Re-write complete\n",
dev_name);
} else {
/* parity is wrong, and is not being fixed.
Exit w/ an error. */
exit(1);
}
}
}
static void
rebuild_in_place(int fd, char *component)
{
RF_SingleComponent_t comp;
int component_num;
int num_cols;
get_component_number(fd, component, &component_num, &num_cols);
comp.row = 0;
comp.column = component_num;
strncpy(comp.component_name, component, sizeof(comp.component_name));
do_ioctl( fd, RAIDFRAME_REBUILD_IN_PLACE, &comp,
"RAIDFRAME_REBUILD_IN_PLACE");
if (verbose) {
printf("Reconstruction status:\n");
sleep(3); /* XXX give reconstruction a chance to start */
do_meter(fd,RAIDFRAME_CHECK_RECON_STATUS_EXT);
}
}
static void
rf_fail_disk(int fd, char *component_to_fail, int do_recon)
{
struct rf_recon_req recon_request;
int component_num;
int num_cols;
get_component_number(fd, component_to_fail, &component_num, &num_cols);
recon_request.row = component_num / num_cols;
recon_request.col = component_num % num_cols;
if (do_recon) {
recon_request.flags = RF_FDFLAGS_RECON;
} else {
recon_request.flags = RF_FDFLAGS_NONE;
}
do_ioctl(fd, RAIDFRAME_FAIL_DISK, &recon_request,
"RAIDFRAME_FAIL_DISK");
if (do_recon && verbose) {
printf("Reconstruction status:\n");
sleep(3); /* XXX give reconstruction a chance to start */
do_meter(fd,RAIDFRAME_CHECK_RECON_STATUS_EXT);
}
}
int
main(int argc,char *argv[])
{
int ch;
int num_options;
unsigned long action;
char config_filename[PATH_MAX];
char dev_name[PATH_MAX];
char name[PATH_MAX];
char component[PATH_MAX];
char autoconf[10];
int do_output;
int do_recon;
int do_rewrite;
int is_clean;
int raidID;
int serial_number;
struct stat st;
int fd;
int force;
int openmode;
num_options = 0;
action = 0;
do_output = 0;
do_recon = 0;
do_rewrite = 0;
is_clean = 0;
serial_number = 0;
force = 0;
openmode = O_RDWR; /* default to read/write */
while ((ch = getopt(argc, argv, "a:A:Bc:C:f:F:g:GiI:l:r:R:sSpPuv"))
!= -1)
switch(ch) {
case 'a':
action = RAIDFRAME_ADD_HOT_SPARE;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'A':
action = RAIDFRAME_SET_AUTOCONFIG;
strlcpy(autoconf, optarg, sizeof(autoconf));
num_options++;
break;
case 'B':
action = RAIDFRAME_COPYBACK;
num_options++;
break;
case 'c':
action = RAIDFRAME_CONFIGURE;
strlcpy(config_filename, optarg,
sizeof(config_filename));
force = 0;
num_options++;
break;
case 'C':
strlcpy(config_filename, optarg,
sizeof(config_filename));
action = RAIDFRAME_CONFIGURE;
force = 1;
num_options++;
break;
case 'f':
action = RAIDFRAME_FAIL_DISK;
strlcpy(component, optarg, sizeof(component));
do_recon = 0;
num_options++;
break;
case 'F':
action = RAIDFRAME_FAIL_DISK;
strlcpy(component, optarg, sizeof(component));
do_recon = 1;
num_options++;
break;
case 'g':
action = RAIDFRAME_GET_COMPONENT_LABEL;
strlcpy(component, optarg, sizeof(component));
openmode = O_RDONLY;
num_options++;
break;
case 'G':
action = RAIDFRAME_GET_INFO;
openmode = O_RDONLY;
do_output = 1;
num_options++;
break;
case 'i':
action = RAIDFRAME_REWRITEPARITY;
num_options++;
break;
case 'I':
action = RAIDFRAME_INIT_LABELS;
serial_number = atoi(optarg);
num_options++;
break;
case 'l':
action = RAIDFRAME_SET_COMPONENT_LABEL;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'r':
action = RAIDFRAME_REMOVE_HOT_SPARE;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'R':
strlcpy(component, optarg, sizeof(component));
action = RAIDFRAME_REBUILD_IN_PLACE;
num_options++;
break;
case 's':
action = RAIDFRAME_GET_INFO;
openmode = O_RDONLY;
num_options++;
break;
case 'S':
action = RAIDFRAME_CHECK_RECON_STATUS_EXT;
openmode = O_RDONLY;
num_options++;
break;
case 'p':
action = RAIDFRAME_CHECK_PARITY;
openmode = O_RDONLY;
num_options++;
break;
case 'P':
action = RAIDFRAME_CHECK_PARITY;
do_rewrite = 1;
num_options++;
break;
case 'u':
action = RAIDFRAME_SHUTDOWN;
num_options++;
break;
case 'v':
verbose = 1;
/* Don't bump num_options, as '-v' is not
an option like the others */
/* num_options++; */
break;
default:
usage();
}
argc -= optind;
argv += optind;
if ((num_options > 1) || (argc == 0))
usage();
strlcpy(name, argv[0], sizeof(name));
fd = opendisk(name, openmode, dev_name, sizeof(dev_name), 0);
if (fd == -1) {
fprintf(stderr, "%s: unable to open device file: %s\n",
getprogname(), name);
exit(1);
}
if (fstat(fd, &st) != 0) {
fprintf(stderr,"%s: stat failure on: %s\n",
getprogname(), dev_name);
exit(1);
}
if (!S_ISBLK(st.st_mode) && !S_ISCHR(st.st_mode)) {
fprintf(stderr,"%s: invalid device: %s\n",
getprogname(), dev_name);
exit(1);
}
raidID = DISKUNIT(st.st_rdev);
switch(action) {
case RAIDFRAME_ADD_HOT_SPARE:
add_hot_spare(fd, component);
break;
case RAIDFRAME_REMOVE_HOT_SPARE:
remove_hot_spare(fd, component);
break;
case RAIDFRAME_CONFIGURE:
rf_configure(fd, config_filename, force);
break;
case RAIDFRAME_SET_AUTOCONFIG:
set_autoconfig(fd, raidID, autoconf);
break;
case RAIDFRAME_COPYBACK:
printf("Copyback.\n");
do_ioctl(fd, RAIDFRAME_COPYBACK, NULL, "RAIDFRAME_COPYBACK");
if (verbose) {
sleep(3); /* XXX give the copyback a chance to start */
printf("Copyback status:\n");
do_meter(fd,RAIDFRAME_CHECK_COPYBACK_STATUS_EXT);
}
break;
case RAIDFRAME_FAIL_DISK:
rf_fail_disk(fd, component, do_recon);
break;
case RAIDFRAME_SET_COMPONENT_LABEL:
set_component_label(fd, component);
break;
case RAIDFRAME_GET_COMPONENT_LABEL:
get_component_label(fd, component);
break;
case RAIDFRAME_INIT_LABELS:
init_component_labels(fd, serial_number);
break;
case RAIDFRAME_REWRITEPARITY:
printf("Initiating re-write of parity\n");
do_ioctl(fd, RAIDFRAME_REWRITEPARITY, NULL,
"RAIDFRAME_REWRITEPARITY");
if (verbose) {
sleep(3); /* XXX give it time to get started */
printf("Parity Re-write status:\n");
do_meter(fd, RAIDFRAME_CHECK_PARITYREWRITE_STATUS_EXT);
}
break;
case RAIDFRAME_CHECK_RECON_STATUS_EXT:
check_status(fd,1);
break;
case RAIDFRAME_GET_INFO:
if (do_output)
rf_output_configuration(fd, dev_name);
else
rf_get_device_status(fd);
break;
case RAIDFRAME_REBUILD_IN_PLACE:
rebuild_in_place(fd, component);
break;
case RAIDFRAME_CHECK_PARITY:
check_parity(fd, do_rewrite, dev_name);
break;
case RAIDFRAME_SHUTDOWN:
do_ioctl(fd, RAIDFRAME_SHUTDOWN, NULL, "RAIDFRAME_SHUTDOWN");
break;
default:
break;
}
close(fd);
exit(0);
}
int
main(int argc,char *argv[])
{
int ch, i;
int num_options;
unsigned long action;
char config_filename[PATH_MAX];
char dev_name[PATH_MAX];
char name[PATH_MAX];
char component[PATH_MAX];
char autoconf[10];
char *parityconf = NULL;
int parityparams[3];
int do_output;
int do_recon;
int do_rewrite;
int raidID;
int serial_number;
struct stat st;
int fd;
int force;
int openmode;
int last_unit;
num_options = 0;
action = 0;
do_output = 0;
do_recon = 0;
do_rewrite = 0;
serial_number = 0;
force = 0;
last_unit = 0;
openmode = O_RDWR; /* default to read/write */
while ((ch = getopt(argc, argv, "a:A:Bc:C:f:F:g:GiI:l:mM:r:R:sSpPuU:v"))
!= -1)
switch(ch) {
case 'a':
action = RAIDFRAME_ADD_HOT_SPARE;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'A':
action = RAIDFRAME_SET_AUTOCONFIG;
strlcpy(autoconf, optarg, sizeof(autoconf));
num_options++;
break;
case 'B':
action = RAIDFRAME_COPYBACK;
num_options++;
break;
case 'c':
action = RAIDFRAME_CONFIGURE;
strlcpy(config_filename, optarg,
sizeof(config_filename));
force = 0;
num_options++;
break;
case 'C':
strlcpy(config_filename, optarg,
sizeof(config_filename));
action = RAIDFRAME_CONFIGURE;
force = 1;
num_options++;
break;
case 'f':
action = RAIDFRAME_FAIL_DISK;
strlcpy(component, optarg, sizeof(component));
do_recon = 0;
num_options++;
break;
case 'F':
action = RAIDFRAME_FAIL_DISK;
strlcpy(component, optarg, sizeof(component));
do_recon = 1;
num_options++;
break;
case 'g':
action = RAIDFRAME_GET_COMPONENT_LABEL;
strlcpy(component, optarg, sizeof(component));
openmode = O_RDONLY;
num_options++;
break;
case 'G':
action = RAIDFRAME_GET_INFO;
openmode = O_RDONLY;
do_output = 1;
num_options++;
break;
case 'i':
action = RAIDFRAME_REWRITEPARITY;
num_options++;
break;
case 'I':
action = RAIDFRAME_INIT_LABELS;
serial_number = xstrtouint(optarg);
num_options++;
break;
case 'm':
action = RAIDFRAME_PARITYMAP_STATUS;
openmode = O_RDONLY;
num_options++;
break;
case 'M':
action = RAIDFRAME_PARITYMAP_SET_DISABLE;
parityconf = strdup(optarg);
num_options++;
/* XXXjld: should rf_pm_configure do the strtol()s? */
i = 0;
while (i < 3 && optind < argc &&
isdigit((int)argv[optind][0]))
parityparams[i++] = xstrtouint(argv[optind++]);
while (i < 3)
parityparams[i++] = 0;
break;
case 'l':
action = RAIDFRAME_SET_COMPONENT_LABEL;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'r':
action = RAIDFRAME_REMOVE_HOT_SPARE;
strlcpy(component, optarg, sizeof(component));
num_options++;
break;
case 'R':
strlcpy(component, optarg, sizeof(component));
action = RAIDFRAME_REBUILD_IN_PLACE;
num_options++;
break;
case 's':
action = RAIDFRAME_GET_INFO;
openmode = O_RDONLY;
num_options++;
break;
case 'S':
action = RAIDFRAME_CHECK_RECON_STATUS_EXT;
openmode = O_RDONLY;
num_options++;
break;
case 'p':
action = RAIDFRAME_CHECK_PARITY;
openmode = O_RDONLY;
num_options++;
break;
case 'P':
action = RAIDFRAME_CHECK_PARITY;
do_rewrite = 1;
num_options++;
break;
case 'u':
action = RAIDFRAME_SHUTDOWN;
num_options++;
break;
case 'U':
action = RAIDFRAME_SET_LAST_UNIT;
num_options++;
last_unit = atoi(optarg);
if (last_unit < 0)
errx(1, "Bad last unit %s", optarg);
break;
case 'v':
verbose = 1;
/* Don't bump num_options, as '-v' is not
an option like the others */
/* num_options++; */
break;
default:
usage();
}
argc -= optind;
argv += optind;
if ((num_options > 1) || (argc == 0))
usage();
if (prog_init && prog_init() == -1)
err(1, "init failed");
strlcpy(name, argv[0], sizeof(name));
fd = opendisk1(name, openmode, dev_name, sizeof(dev_name), 0,
prog_open);
if (fd == -1)
err(1, "Unable to open device file: %s", name);
if (prog_fstat(fd, &st) == -1)
err(1, "stat failure on: %s", dev_name);
if (!S_ISBLK(st.st_mode) && !S_ISCHR(st.st_mode))
err(1, "invalid device: %s", dev_name);
raidID = DISKUNIT(st.st_rdev);
switch(action) {
case RAIDFRAME_ADD_HOT_SPARE:
add_hot_spare(fd, component);
break;
case RAIDFRAME_REMOVE_HOT_SPARE:
remove_hot_spare(fd, component);
break;
case RAIDFRAME_CONFIGURE:
rf_configure(fd, config_filename, force);
break;
case RAIDFRAME_SET_AUTOCONFIG:
set_autoconfig(fd, raidID, autoconf);
break;
case RAIDFRAME_COPYBACK:
printf("Copyback.\n");
do_ioctl(fd, RAIDFRAME_COPYBACK, NULL, "RAIDFRAME_COPYBACK");
if (verbose) {
sleep(3); /* XXX give the copyback a chance to start */
printf("Copyback status:\n");
do_meter(fd,RAIDFRAME_CHECK_COPYBACK_STATUS_EXT);
}
break;
case RAIDFRAME_FAIL_DISK:
rf_fail_disk(fd, component, do_recon);
break;
case RAIDFRAME_SET_COMPONENT_LABEL:
set_component_label(fd, component);
break;
case RAIDFRAME_GET_COMPONENT_LABEL:
get_component_label(fd, component);
break;
case RAIDFRAME_INIT_LABELS:
init_component_labels(fd, serial_number);
break;
case RAIDFRAME_REWRITEPARITY:
printf("Initiating re-write of parity\n");
do_ioctl(fd, RAIDFRAME_REWRITEPARITY, NULL,
"RAIDFRAME_REWRITEPARITY");
if (verbose) {
sleep(3); /* XXX give it time to get started */
printf("Parity Re-write status:\n");
do_meter(fd, RAIDFRAME_CHECK_PARITYREWRITE_STATUS_EXT);
}
break;
case RAIDFRAME_CHECK_RECON_STATUS_EXT:
check_status(fd,1);
break;
case RAIDFRAME_GET_INFO:
if (do_output)
rf_output_configuration(fd, dev_name);
else
rf_get_device_status(fd);
break;
case RAIDFRAME_PARITYMAP_STATUS:
rf_output_pmstat(fd, raidID);
break;
case RAIDFRAME_PARITYMAP_SET_DISABLE:
rf_pm_configure(fd, raidID, parityconf, parityparams);
break;
case RAIDFRAME_REBUILD_IN_PLACE:
rebuild_in_place(fd, component);
break;
case RAIDFRAME_CHECK_PARITY:
check_parity(fd, do_rewrite, dev_name);
break;
case RAIDFRAME_SHUTDOWN:
do_ioctl(fd, RAIDFRAME_SHUTDOWN, NULL, "RAIDFRAME_SHUTDOWN");
break;
case RAIDFRAME_SET_LAST_UNIT:
do_ioctl(fd, RAIDFRAME_SET_LAST_UNIT, &last_unit,
"RAIDFRAME_SET_LAST_UNIT");
break;
default:
break;
}
prog_close(fd);
exit(0);
}
