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 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;
int use_sudo=0;
const list_disk_t *element_disk;
const list_disk_t *current_disk;
static 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;
#ifdef HAVE_NCURSES
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):");
#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);
}
}
{
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 incorrect size, check HD jumper settings, BIOS");
wmove(stdscr,line,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)
options="OSQ";
else
options="PNOSQ";
}
else
{
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) &&
interface_partition_type(disk, verbose, current_cmd)==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;
}
}
