int do_curses_photorec(struct ph_param *params, struct ph_options *options, const list_disk_t *list_disk)
{
static alloc_data_t list_search_space={
.list = TD_LIST_HEAD_INIT(list_search_space.list)
};
if(params->cmd_device==NULL)
{
char *saved_device=NULL;
char *saved_cmd=NULL;
session_load(&saved_device, &saved_cmd,&list_search_space);
if(saved_device!=NULL && saved_cmd!=NULL && !td_list_empty(&list_search_space.list)
#ifdef HAVE_NCURSES
&& ask_confirmation("Continue previous session ? (Y/N)")!=0
#endif
)
{
/* yes */
params->cmd_run=saved_cmd;
params->cmd_device=saved_device;
}
else
{
free(saved_device);
free(saved_cmd);
free_list_search_space(&list_search_space);
}
}
if(params->cmd_device!=NULL && params->cmd_run!=NULL)
{
params->disk=photorec_disk_selection_cli(params->cmd_device, list_disk, &list_search_space);
#ifdef HAVE_NCURSES
if(params->disk==NULL)
{
log_critical("No disk found\n");
return intrf_no_disk_ncurses("PhotoRec");
}
if(change_arch_type_cli(params->disk, options->verbose, ¶ms->cmd_run)==0 ||
change_arch_type_ncurses(params->disk, options->verbose)==0)
{
menu_photorec(params, options, &list_search_space);
}
return 0;
#else
if(params->disk==NULL)
{
log_critical("No disk found\n");
return 0;
}
change_arch_type_cli(params->disk, options->verbose, ¶ms->cmd_run);
menu_photorec(params, options, &list_search_space);
return 0;
#endif
}
#ifdef HAVE_NCURSES
return photorec_disk_selection_ncurses(params, options, list_disk, &list_search_space);
#else
return 0;
#endif
}
static int testdisk_disk_selection_cli(int verbose,int dump_ind, const list_disk_t *list_disk, const int saveheader, const char *cmd_device, char **current_cmd)
{
const list_disk_t *element_disk;
const list_disk_t *current_disk=NULL;
if(cmd_device!=NULL)
{
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
if(strcmp(element_disk->disk->device,cmd_device)==0)
current_disk=element_disk;
}
}
else
current_disk=list_disk;
if(current_disk==NULL)
{
log_critical("No disk found\n");
return 0;
}
if(*current_cmd!=NULL)
{
while(*current_cmd[0]==',')
(*current_cmd)++;
{
disk_t *disk=current_disk->disk;
autodetect_arch(disk, NULL);
autoset_unit(disk);
if(interface_check_disk_capacity(disk)==0 &&
interface_check_disk_access(disk, current_cmd)==0 &&
(change_arch_type_cli(disk, verbose, current_cmd)==0
#ifdef HAVE_NCURSES
|| change_arch_type_ncurses(disk, verbose)==0
#endif
))
{
menu_disk(disk, verbose, dump_ind, saveheader, current_cmd);
}
}
}
return 0;
}
static int photorec_disk_selection_ncurses(struct ph_param *params, struct ph_options *options, const list_disk_t *list_disk, alloc_data_t *list_search_space)
{
int command;
int real_key;
unsigned int menu=0;
int offset=0;
int pos_num=0;
int use_sudo=0;
const list_disk_t *element_disk;
const list_disk_t *current_disk=list_disk;
static const struct MenuItem menuMain[]=
{
{ 'P', "Previous",""},
{ 'N', "Next","" },
{ 'O',"Proceed",""},
{ 'S', "Sudo", "Use the sudo command to restart as root"},
{ 'Q',"Quit","Quit program"},
{ 0,NULL,NULL}
};
if(list_disk==NULL)
{
log_critical("No disk found\n");
return intrf_no_disk_ncurses("PhotoRec");
}
/* ncurses interface */
while(1)
{
const char *menu_options;
int i;
aff_copy(stdscr);
wmove(stdscr,4,0);
wprintw(stdscr," PhotoRec is free software, and");
wmove(stdscr,5,0);
wprintw(stdscr,"comes with ABSOLUTELY NO WARRANTY.");
wmove(stdscr,7,0);
wprintw(stdscr,"Select a media (use Arrow keys, then press Enter):");
#if defined(KEY_MOUSE) && defined(ENABLE_MOUSE)
mousemask(ALL_MOUSE_EVENTS, NULL);
#endif
for(i=0,element_disk=list_disk;
element_disk!=NULL && i<offset+NBR_DISK_MAX;
i++, element_disk=element_disk->next)
{
if(i<offset)
continue;
wmove(stdscr,8+i-offset,0);
if(element_disk!=current_disk)
wprintw(stdscr," %s\n",element_disk->disk->description_short(element_disk->disk));
else
{
wattrset(stdscr, A_REVERSE);
wprintw(stdscr,">%s\n",element_disk->disk->description_short(element_disk->disk));
wattroff(stdscr, A_REVERSE);
}
}
{
mvwaddstr(stdscr, INTER_NOTE_Y,0,"Note: ");
#if defined(__CYGWIN__) || defined(__MINGW32__)
#else
#ifndef DJGPP
#ifdef HAVE_GETEUID
if(geteuid()!=0)
{
if(has_colors())
wbkgdset(stdscr,' ' | A_BOLD | COLOR_PAIR(1));
wprintw(stdscr,"Some disks won't appear unless you're root user.");
if(has_colors())
wbkgdset(stdscr,' ' | COLOR_PAIR(0));
#ifdef SUDO_BIN
use_sudo=1;
#endif
}
#endif
#endif
#endif
wmove(stdscr, INTER_NOTE_Y+1, 0);
wprintw(stdscr,"Disk capacity must be correctly detected for a successful recovery.");
wmove(stdscr, INTER_NOTE_Y+2, 0);
wprintw(stdscr,"If a disk listed above has incorrect size, check HD jumper settings, BIOS");
wmove(stdscr, INTER_NOTE_Y+3, 0);
wprintw(stdscr,"detection, and install the latest OS patches and disk drivers.");
}
if(use_sudo > 0)
{
if(i<=NBR_DISK_MAX && element_disk==NULL)
menu_options="OSQ";
else
menu_options="PNOSQ";
}
else
{
if(i<=NBR_DISK_MAX && element_disk==NULL)
menu_options="OQ";
else
menu_options="PNOQ";
}
command = wmenuSelect_ext(stdscr, INTER_NOTE_Y-1, INTER_DISK_Y, INTER_DISK_X, menuMain, 8,
menu_options, MENU_HORIZ | MENU_BUTTON | MENU_ACCEPT_OTHERS, &menu,&real_key);
#if defined(KEY_MOUSE) && defined(ENABLE_MOUSE)
if(command == KEY_MOUSE)
{
MEVENT event;
if(getmouse(&event) == OK)
{ /* When the user clicks left mouse button */
if((event.bstate & BUTTON1_CLICKED) || (event.bstate & BUTTON1_DOUBLE_CLICKED))
{
if(event.y >=8 && event.y<8+NBR_DISK_MAX)
{
const int pos_num_old=pos_num;
/* Disk selection */
while(pos_num > event.y-(8-offset) && current_disk->prev!=NULL)
{
current_disk=current_disk->prev;
pos_num--;
}
while(pos_num < event.y-(8-offset) && current_disk->next!=NULL)
{
current_disk=current_disk->next;
pos_num++;
}
if(((event.bstate & BUTTON1_CLICKED) && pos_num==pos_num_old) ||
(event.bstate & BUTTON1_DOUBLE_CLICKED))
command='O';
}
else
command = menu_to_command(INTER_NOTE_Y-1, INTER_DISK_Y, INTER_DISK_X, menuMain, 8,
menu_options, MENU_HORIZ | MENU_BUTTON | MENU_ACCEPT_OTHERS, event.y, event.x);
}
}
}
#endif
switch(command)
{
case KEY_UP:
case 'P':
if(current_disk->prev!=NULL)
{
current_disk=current_disk->prev;
pos_num--;
}
break;
case KEY_DOWN:
case 'N':
if(current_disk->next!=NULL)
{
current_disk=current_disk->next;
pos_num++;
}
break;
case KEY_PPAGE:
for(i=0;i<NBR_DISK_MAX-1 && current_disk->prev!=NULL;i++)
{
current_disk=current_disk->prev;
pos_num--;
}
break;
case KEY_NPAGE:
for(i=0;i<NBR_DISK_MAX-1 && current_disk->next!=NULL;i++)
{
current_disk=current_disk->next;
pos_num++;
}
break;
case 'o':
case 'O':
{
disk_t *disk=current_disk->disk;
const int hpa_dco=is_hpa_or_dco(disk);
autodetect_arch(disk, &arch_none);
params->disk=disk;
if((hpa_dco==0 || interface_check_hidden_ncurses(disk, hpa_dco)==0) &&
(options->expert == 0 ||
change_arch_type_ncurses(disk, options->verbose)==0))
menu_photorec(params, options, list_search_space);
}
break;
case 's':
case 'S':
return 1;
case 'q':
case 'Q':
return 0;
}
if(pos_num<offset)
offset=pos_num;
if(pos_num>=offset+NBR_DISK_MAX)
offset=pos_num-NBR_DISK_MAX+1;
}
}
static int testdisk_disk_selection_ncurses(int verbose,int dump_ind, const list_disk_t *list_disk, const int saveheader, char **current_cmd)
{
int command='Q';
unsigned int menu=0;
int offset=0;
int pos_num=0;
#ifdef SUDO_BIN
int use_sudo=0;
#endif
const list_disk_t *element_disk;
const list_disk_t *current_disk;
static const struct MenuItem menuMain[]=
{
{ 'P', "Previous", ""},
{ 'N', "Next", "" },
{ 'O', "Proceed", ""},
{ 'S', "Sudo", "Use the sudo command to restart as root"},
{ 'Q', "Quit", "Quit program"},
{ 0, NULL, NULL}
};
if(list_disk==NULL)
{
log_critical("No disk found\n");
return intrf_no_disk_ncurses("TestDisk");
}
current_disk=list_disk;
/* ncurses interface */
while(1)
{
const char *options;
int i;
aff_copy(stdscr);
wmove(stdscr,4,0);
wprintw(stdscr," TestDisk is free software, and");
wmove(stdscr,5,0);
wprintw(stdscr,"comes with ABSOLUTELY NO WARRANTY.");
wmove(stdscr,7,0);
wprintw(stdscr,"Select a media (use Arrow keys, then press Enter):");
for(i=0,element_disk=list_disk;
element_disk!=NULL && i<offset+NBR_DISK_MAX;
i++, element_disk=element_disk->next)
{
if(i<offset)
continue;
wmove(stdscr,8+i-offset,0);
if(element_disk!=current_disk)
wprintw(stdscr," %s\n",element_disk->disk->description_short(element_disk->disk));
else
{
wattrset(stdscr, A_REVERSE);
wprintw(stdscr,">%s\n",element_disk->disk->description_short(element_disk->disk));
wattroff(stdscr, A_REVERSE);
}
}
{
int line=INTER_NOTE_Y;
mvwaddstr(stdscr,line++,0,"Note: ");
#if defined(__CYGWIN__) || defined(__MINGW32__) || defined(DJGPP)
#else
#ifdef HAVE_GETEUID
if(geteuid()!=0)
{
if(has_colors())
wbkgdset(stdscr,' ' | A_BOLD | COLOR_PAIR(1));
waddstr(stdscr,"Some disks won't appear unless you are root user.");
if(has_colors())
wbkgdset(stdscr,' ' | COLOR_PAIR(0));
wmove(stdscr,line++,0);
#ifdef SUDO_BIN
use_sudo=1;
#endif
}
#endif
#endif
waddstr(stdscr,"Disk capacity must be correctly detected for a successful recovery.");
wmove(stdscr,line++,0);
wprintw(stdscr,"If a disk listed above has an incorrect size, check HD jumper settings and BIOS");
wmove(stdscr,line,0);
wprintw(stdscr,"detection, and install the latest OS patches and disk drivers.");
}
#ifdef SUDO_BIN
if(use_sudo > 0)
{
if(i<=NBR_DISK_MAX && element_disk==NULL)
options="OSQ";
else
options="PNOSQ";
}
else
#endif
{
if(i<=NBR_DISK_MAX && element_disk==NULL)
options="OQ";
else
options="PNOQ";
}
command = wmenuSelect_ext(stdscr, INTER_NOTE_Y-1, INTER_DISK_Y, INTER_DISK_X, menuMain, 8,
options, MENU_HORIZ | MENU_BUTTON | MENU_ACCEPT_OTHERS, &menu,NULL);
switch(command)
{
case 'p':
case 'P':
case KEY_UP:
if(current_disk->prev!=NULL)
{
current_disk=current_disk->prev;
pos_num--;
}
break;
case 'n':
case 'N':
case KEY_DOWN:
if(current_disk->next!=NULL)
{
current_disk=current_disk->next;
pos_num++;
}
break;
case KEY_PPAGE:
for(i=0;i<NBR_DISK_MAX-1 && current_disk->prev!=NULL;i++)
{
current_disk=current_disk->prev;
pos_num--;
}
break;
case KEY_NPAGE:
for(i=0;i<NBR_DISK_MAX-1 && current_disk->next!=NULL;i++)
{
current_disk=current_disk->next;
pos_num++;
}
break;
case 'o':
case 'O':
{
disk_t *disk=current_disk->disk;
const int hpa_dco=is_hpa_or_dco(disk);
autodetect_arch(disk, NULL);
autoset_unit(disk);
if(interface_check_disk_capacity(disk)==0 &&
interface_check_disk_access(disk, current_cmd)==0 &&
(hpa_dco==0 || interface_check_hidden_ncurses(disk, hpa_dco)==0) &&
change_arch_type_ncurses(disk, verbose)==0)
{
if(menu_disk(disk, verbose, dump_ind, saveheader, current_cmd))
return 0;
}
}
break;
case 's':
case 'S':
return 1;
case 'q':
case 'Q':
return 0;
}
if(pos_num<offset)
offset=pos_num;
if(pos_num>=offset+NBR_DISK_MAX)
offset=pos_num-NBR_DISK_MAX+1;
}
}
