static void Close(DC_RendererCtx *ctx) {
SlidingWindow *priv = ctx->priv;
DC_ProcedureCtx *actctx = ctx->procedure_ctx;
priv->order_hangup = 1;
pthread_join(priv->render_thread, NULL);
if (actctx->interrupt)
wprintw(priv->summary, "Aborted.\n");
else
wprintw(priv->summary, "Completed.\n");
wprintw(priv->summary, "Press 'm' for menu");
wrefresh(priv->summary);
beep();
while (getchar() != 'm')
;
delwin(priv->legend);
delwin(priv->access_time_stats);
delwin(priv->vis);
delwin(priv->avg_speed);
delwin(priv->eta);
delwin(priv->summary);
delwin(priv->w_end_lba);
delwin(priv->w_cur_lba);
clear_body();
}
static DC_Dev *menu_choose_device(DC_DevList *devlist) {
int devs_num = dc_dev_list_size(devlist);
if (devs_num == 0) {
dialog_msgbox("Info", "No devices found", 0, 0, 1);
return NULL;
}
char *items[2 * devs_num];
int i;
for (i = 0; i < devs_num; i++) {
DC_Dev *dev = dc_dev_list_get_entry(devlist, i);
char dev_descr_buf[80];
ui_dev_descr_format(dev_descr_buf, sizeof(dev_descr_buf), dev);
items[2*i] = dev->dev_fs_name;
items[2*i+1] = strdup(dev_descr_buf);
}
clear_body();
dialog_vars.no_items = 0;
dialog_vars.item_help = 0;
dialog_vars.input_result = NULL;
dialog_vars.default_button = 0; // Focus on "OK"
int ret = dialog_menu("Choose device", "", 0, 0, 0, devs_num, items);
for (i = 0; i < devs_num; i++)
free(items[2*i+1]);
if (ret != 0)
return NULL;
for (i = 0; i < devs_num; i++) {
DC_Dev *dev = dc_dev_list_get_entry(devlist, i);
if (!strcmp(dev->dev_fs_name, dialog_vars.input_result))
return dev;
}
assert(0);
return NULL;
}
static int global_init(void) {
int r;
// TODO check all retcodes
setlocale(LC_ALL, "");
initscr();
init_dialog(stdin, stdout);
dialog_vars.item_help = 0;
start_color();
init_my_colors();
noecho();
cbreak();
scrollok(stdscr, FALSE);
keypad(stdscr, TRUE);
clear_body();
// init libdevcheck
r = dc_init();
assert(!r);
RENDERER_REGISTER(sliding_window);
RENDERER_REGISTER(whole_space);
dc_log_set_callback(log_cb, NULL);
r = atexit(global_fini);
assert(r == 0);
return 0;
}
static void Close(DC_RendererCtx *ctx) {
WholeSpace *priv = ctx->priv;
DC_ProcedureCtx *actctx = ctx->procedure_ctx;
priv->order_hangup = 1;
pthread_join(priv->render_thread, NULL);
if (actctx->interrupt)
wprintw(priv->summary, "Aborted.\n");
else
wprintw(priv->summary, "Completed.\n");
wprintw(priv->summary, "Press any key");
wrefresh(priv->summary);
beep();
getch();
delwin(priv->legend);
delwin(priv->vis);
delwin(priv->avg_speed);
delwin(priv->eta);
delwin(priv->summary);
delwin(priv->w_end_lba);
delwin(priv->w_cur_lba);
delwin(priv->w_log);
clear_body();
free(priv->blocks_map);
}
static DC_Procedure *menu_choose_procedure(DC_Dev *dev) {
(void)dev;
int nb_procedures = dc_get_nb_procedures();
const char *items[nb_procedures];
DC_Procedure *procedures[nb_procedures];
int nb_items = 0;
DC_Procedure *procedure = NULL;
while ((procedure = dc_get_next_procedure(procedure))) {
if (!dev->ata_capable && (procedure->flags & DC_PROC_FLAG_REQUIRES_ATA))
continue;
items[nb_items] = procedure->display_name;
procedures[nb_items] = procedure;
nb_items++;
}
while (1) {
clear_body();
dialog_vars.no_items = 1;
dialog_vars.item_help = 0;
dialog_vars.input_result = NULL;
dialog_vars.default_button = 0; // Focus on "OK"
dialog_vars.extra_button = 1;
dialog_vars.extra_label = "Help";
int ret = dialog_menu("Choose procedure", "", 0, 0, 0, nb_items, (/* should be const */char**)items);
dialog_vars.extra_button = 0;
if ((ret != DLG_EXIT_OK) && (ret != DLG_EXIT_EXTRA))
return NULL; // User quit dialog, exit
procedure = NULL;
for (int i = 0; i < nb_items; i++)
if (!strcmp(items[i], dialog_vars.input_result)) {
procedure = procedures[i];
break;
}
assert(procedure);
if (ret == DLG_EXIT_EXTRA) {
dialog_msgbox(procedure->display_name, procedure->help ? : "No help", 0, 0, 1);
continue;
}
assert(ret == DLG_EXIT_OK);
return procedure;
}
int main() {
int r;
r = global_init();
if (r) {
fprintf(stderr, "init fail\n");
return r;
}
// get list of devices
DC_DevList *devlist = dc_dev_list();
assert(devlist);
while (1) {
// draw menu of device choice
DC_Dev *chosen_dev = menu_choose_device(devlist);
if (!chosen_dev) {
break;
}
// draw procedures menu
DC_Procedure *act = menu_choose_procedure(chosen_dev);
if (!act)
continue;
if (act->flags & DC_PROC_FLAG_INVASIVE) {
char *ask;
r = asprintf(&ask, "This operation is invasive, i.e. it may make your data unreachable or even destroy it completely. Are you sure you want to proceed it on %s (%s)?",
chosen_dev->dev_fs_name, chosen_dev->model_str);
assert(r != -1);
dialog_vars.default_button = 1; // Focus on "No"
r = dialog_yesno("Confirmation", ask, 0, 0);
// Yes = 0 (FALSE), No = 1, Escape = -1
free(ask);
if (/* No */ r)
continue;
if (chosen_dev->mounted) {
dialog_vars.default_button = 1; // Focus on "No"
r = dialog_yesno("Confirmation", "This disk is mounted. Are you really sure you want to proceed?", 0, 0);
if (r)
continue;
}
}
DC_OptionSetting *option_set = calloc(act->options_num + 1, sizeof(DC_OptionSetting));
int i;
r = 0;
for (i = 0; i < act->options_num; i++) {
option_set[i].name = act->options[i].name;
r = act->suggest_default_value(chosen_dev, &option_set[i]);
if (r) {
dc_log(DC_LOG_ERROR, "Failed to get default value suggestion on '%s'", option_set[i].name);
break;
}
r = ask_option_value(act, &option_set[i], &act->options[i]);
if (r)
break;
}
if (r)
continue;
// Show relaxing banner when copying with journal. Copy journal processing takes some time.
if (!strcmp(act->name, "copy")) {
int uses_journal = 0;
for (i = 0; i < act->options_num; i++) {
if (!strcmp(option_set[i].name, "use_journal")) {
uses_journal = 1;
break;
}
}
if (uses_journal)
dialog_msgbox("Info", "Please wait while operation journal is processed", 0, 0, 0 /* non-pausing */);
}
clear_body();
DC_ProcedureCtx *actctx;
r = dc_procedure_open(act, chosen_dev, &actctx, option_set);
if (r) {
dialog_msgbox("Error", "Procedure init fail", 0, 0, 1);
continue;
}
if (!act->perform)
continue;
DC_Renderer *renderer;
if (!strcmp(act->name, "copy"))
renderer = dc_find_renderer("whole_space");
else
renderer = dc_find_renderer("sliding_window");
render_procedure(actctx, renderer);
} // while(1)
return 0;
}