void change_part_type(const disk_t *disk_car,partition_t *partition, char **current_cmd)
{
if(*current_cmd!=NULL)
{
change_part_type_cli(disk_car, partition, current_cmd);
return;
}
if(partition->arch==NULL)
{
log_error("change_part_type arch==NULL\n");
return;
}
if(partition->arch==&arch_gpt)
{
gpt_change_part_type(disk_car, partition);
log_info("Change partition type:\n");
log_partition(disk_car,partition);
partition->arch=&arch_none;
change_part_type_ncurses2(disk_car, partition);
log_info("Change partition type:\n");
log_partition(disk_car,partition);
partition->arch=&arch_gpt;
return ;
}
if(partition->arch->set_part_type==NULL)
{
log_error("change_part_type set_part_type==NULL\n");
return;
}
if(partition->arch==&arch_i386 || partition->arch==&arch_sun)
change_part_type_ncurses(disk_car, partition);
else
change_part_type_ncurses2(disk_car, partition);
log_info("Change partition type:\n");
log_partition(disk_car,partition);
}
list_part_t *add_partition_gpt_cli(disk_t *disk_car,list_part_t *list_part, char **current_cmd)
{
partition_t *new_partition=partition_new(&arch_gpt);
new_partition->part_offset=disk_car->sector_size;
new_partition->part_size=disk_car->disk_size-new_partition->part_offset;
while(*current_cmd[0]==',')
(*current_cmd)++;
while(1)
{
if(strncmp(*current_cmd,"s,",2)==0)
{
uint64_t part_offset;
(*current_cmd)+=2;
part_offset=new_partition->part_offset;
new_partition->part_offset=(uint64_t)ask_number_cli(
current_cmd,
new_partition->part_offset/disk_car->sector_size,
1,
(disk_car->disk_size-1)/disk_car->sector_size,
"Enter the starting sector ") *
(uint64_t)disk_car->sector_size;
new_partition->part_size=new_partition->part_size + part_offset - new_partition->part_offset;
}
else if(strncmp(*current_cmd,"S,",2)==0)
{
(*current_cmd)+=2;
new_partition->part_size=(uint64_t)ask_number_cli(
current_cmd,
(new_partition->part_offset+new_partition->part_size-1)/disk_car->sector_size,
new_partition->part_offset/disk_car->sector_size,
(disk_car->disk_size-1)/disk_car->sector_size,
"Enter the ending sector ") *
(uint64_t)disk_car->sector_size +
disk_car->sector_size - new_partition->part_offset;
}
else if(strncmp(*current_cmd,"T,",2)==0)
{
(*current_cmd)+=2;
change_part_type_cli(disk_car,new_partition,current_cmd);
}
else if(new_partition->part_size>0 && guid_cmp(new_partition->part_type_gpt, GPT_ENT_TYPE_UNUSED)!=0)
{
int insert_error=0;
list_part_t *new_list_part=insert_new_partition(list_part, new_partition, 0, &insert_error);
if(insert_error>0)
{
free(new_partition);
return new_list_part;
}
new_partition->status=STATUS_PRIM;
if(test_structure_gpt(list_part)!=0)
new_partition->status=STATUS_DELETED;
return new_list_part;
}
else
{
free(new_partition);
return list_part;
}
}
}
void interface_adv(disk_t *disk_car, const int verbose,const int dump_ind, const unsigned int expert, char**current_cmd)
{
#ifdef HAVE_NCURSES
int offset=0;
int current_element_num=0;
int old_LINES=LINES;
#endif
int rewrite=1;
unsigned int menu=0;
list_part_t *element;
list_part_t *list_part;
list_part_t *current_element;
log_info("\nInterface Advanced\n");
list_part=disk_car->arch->read_part(disk_car,verbose,0);
current_element=list_part;
log_all_partitions(disk_car, list_part);
while(1)
{
const char *options;
static struct MenuItem menuAdv[]=
{
{'t',"Type","Change type, this setting will not be saved on disk"},
{'b',"Boot","Boot sector recovery"},
{'s',"Superblock",NULL},
{'l',"List", "List and copy files"},
{'u',"Undelete", "File undelete"},
{'c',"Image Creation", "Create an image"},
// {'a',"Add", "Add temporary partition (Expert only)"},
{'q',"Quit","Return to main menu"},
{0,NULL,NULL}
};
int command;
#ifdef HAVE_NCURSES
int i;
if(old_LINES!=LINES)
{
old_LINES=LINES;
rewrite=1;
}
if(rewrite!=0)
{
aff_copy(stdscr);
wmove(stdscr,4,0);
wprintw(stdscr,"%s",disk_car->description(disk_car));
if(list_part!=NULL)
mvwaddstr(stdscr,6,0,msg_PART_HEADER_LONG);
rewrite=0;
}
for(i=0,element=list_part; element!=NULL && i<offset+INTER_ADV; element=element->next,i++)
{
if(i<offset)
continue;
wmove(stdscr,7+i-offset,0);
wclrtoeol(stdscr); /* before addstr for BSD compatibility */
if(element==current_element)
{
wattrset(stdscr, A_REVERSE);
waddstr(stdscr, ">");
aff_part(stdscr,AFF_PART_ORDER|AFF_PART_STATUS,disk_car,element->part);
wattroff(stdscr, A_REVERSE);
} else
{
waddstr(stdscr, " ");
aff_part(stdscr,AFF_PART_ORDER|AFF_PART_STATUS,disk_car,element->part);
}
}
wmove(stdscr,7+INTER_ADV,5);
wclrtoeol(stdscr);
if(element!=NULL)
wprintw(stdscr, "Next");
#endif
if(current_element==NULL)
{
options="q";
#ifdef HAVE_NCURSES
wmove(stdscr,7,0);
wattrset(stdscr, A_REVERSE);
wprintw(stdscr,"No partition available.");
wattroff(stdscr, A_REVERSE);
#endif
}
else
{
const partition_t *partition=current_element->part;
if(menu==0 && (disk_car->arch!=&arch_none || partition->upart_type!=UP_UNK))
menu=1;
if(is_part_fat(partition))
{
options="tubcq";
menu=(partition->upart_type==UP_UNK?1:4);
}
else if(is_part_ntfs(partition))
options="tlubcq";
else if(is_part_linux(partition))
{
if(partition->upart_type==UP_EXT2)
options="tuscq";
else
options="tlscq";
menuAdv[2].desc="Locate ext2/ext3/ext4 backup superblock";
}
else if(is_part_hfs(partition) || is_part_hfsp(partition))
{
options="tscq";
menuAdv[2].desc="Locate HFS/HFS+ backup volume header";
}
else if(is_fat(partition))
options="tubcq";
else if(is_ntfs(partition))
options="tlubcq";
else if(is_exfat(partition))
options="tlubcq";
else if(is_linux(partition))
{
if(partition->upart_type==UP_EXT2)
options="tluscq";
else
options="tlscq";
menuAdv[2].desc="Locate ext2/ext3/ext4 backup superblock";
}
else if(is_hfs(partition) || is_hfsp(partition))
{
options="tscq";
menuAdv[2].desc="Locate HFS/HFS+ backup volume header";
}
else
options="tcq";
}
if(*current_cmd!=NULL)
{
int keep_asking;
command='q';
do
{
keep_asking=0;
while(*current_cmd[0]==',')
(*current_cmd)++;
if(strncmp(*current_cmd,"type",4)==0)
{
(*current_cmd)+=4;
command='t';
}
else if(strncmp(*current_cmd,"addpart",7)==0)
{
(*current_cmd)+=7;
command='a';
}
else if(strncmp(*current_cmd,"boot",4)==0)
{
(*current_cmd)+=4;
command='b';
}
else if(strncmp(*current_cmd,"copy",4)==0)
{
(*current_cmd)+=4;
command='c';
}
else if(strncmp(*current_cmd,"list",4)==0)
{
(*current_cmd)+=4;
command='l';
}
else if(strncmp(*current_cmd,"undelete",8)==0)
{
(*current_cmd)+=8;
command='u';
}
else if(strncmp(*current_cmd,"superblock",10)==0)
{
(*current_cmd)+=10;
command='s';
}
else if(isdigit(*current_cmd[0]))
{
const unsigned int order= atoi(*current_cmd);
while(*current_cmd[0]!=',' && *current_cmd[0]!='\0')
(*current_cmd)++;
for(element=list_part; element!=NULL && element->part->order!=order; element=element->next);
if(element!=NULL)
{
current_element=element;
keep_asking=1;
}
}
} while(keep_asking>0);
}
else
{
#ifdef HAVE_NCURSES
command = wmenuSelect_ext(stdscr, INTER_ADV_Y+1, INTER_ADV_Y, INTER_ADV_X, menuAdv, 8, options,
MENU_HORIZ | MENU_BUTTON | MENU_ACCEPT_OTHERS, &menu, NULL);
#else
command = 'q';
#endif
}
switch(command)
{
case 'q':
case 'Q':
part_free_list(list_part);
return;
#ifdef HAVE_NCURSES
case 'a':
case 'A':
if(disk_car->arch!=&arch_none)
{
if(*current_cmd!=NULL)
list_part=add_partition_cli(disk_car, list_part, current_cmd);
else
list_part=add_partition_ncurses(disk_car, list_part);
current_element=list_part;
rewrite=1;
}
break;
#endif
}
#ifdef HAVE_NCURSES
if(current_element!=NULL)
{
switch(command)
{
case 'p':
case 'P':
case KEY_UP:
if(current_element->prev!=NULL)
{
current_element=current_element->prev;
current_element_num--;
}
break;
case 'n':
case 'N':
case KEY_DOWN:
if(current_element->next!=NULL)
{
current_element=current_element->next;
current_element_num++;
}
break;
case KEY_PPAGE:
for(i=0; i<INTER_ADV-1 && current_element->prev!=NULL; i++)
{
current_element=current_element->prev;
current_element_num--;
}
break;
case KEY_NPAGE:
for(i=0; i<INTER_ADV-1 && current_element->next!=NULL; i++)
{
current_element=current_element->next;
current_element_num++;
}
break;
case 'b':
case 'B':
{
partition_t *partition=current_element->part;
if(is_part_fat32(partition))
{
fat32_boot_sector(disk_car, partition, verbose, dump_ind, expert,current_cmd);
rewrite=1;
}
else if(is_part_fat12(partition) || is_part_fat16(partition))
{
fat1x_boot_sector(disk_car, partition, verbose, dump_ind,expert,current_cmd);
rewrite=1;
}
else if(is_part_ntfs(partition))
{
if(partition->upart_type==UP_EXFAT)
exFAT_boot_sector(disk_car, partition, verbose, current_cmd);
else
ntfs_boot_sector(disk_car, partition, verbose, expert, current_cmd);
rewrite=1;
}
else if(partition->upart_type==UP_FAT32)
{
fat32_boot_sector(disk_car, partition, verbose, dump_ind, expert,current_cmd);
rewrite=1;
}
else if(partition->upart_type==UP_FAT12 || partition->upart_type==UP_FAT16)
{
fat1x_boot_sector(disk_car, partition, verbose, dump_ind,expert,current_cmd);
rewrite=1;
}
else if(partition->upart_type==UP_NTFS)
{
ntfs_boot_sector(disk_car, partition, verbose, expert, current_cmd);
rewrite=1;
}
else if(partition->upart_type==UP_EXFAT)
{
exFAT_boot_sector(disk_car, partition, verbose, current_cmd);
rewrite=1;
}
}
break;
case 'c':
case 'C':
{
char *dst_path;
#ifdef HAVE_NCURSES
if(*current_cmd!=NULL)
dst_path=get_default_location();
else
{
char msg[256];
snprintf(msg, sizeof(msg),
"Please select where to store the file image.dd (%u MB), an image of the partition",
(unsigned int)(current_element->part->part_size/1000/1000));
dst_path=ask_location(msg, "", NULL);
}
#else
dst_path=get_default_location();
#endif
if(dst_path!=NULL)
{
char *filename=(char *)MALLOC(strlen(dst_path) + 1 + strlen(DEFAULT_IMAGE_NAME) + 1);
strcpy(filename, dst_path);
strcat(filename, "/");
strcat(filename, DEFAULT_IMAGE_NAME);
disk_image(disk_car, current_element->part, filename);
free(filename);
free(dst_path);
}
}
rewrite=1;
break;
case 'u':
case 'U':
{
partition_t *partition=current_element->part;
if(partition->sb_offset!=0 && partition->sb_size>0)
{
io_redir_add_redir(disk_car,
partition->part_offset+partition->sborg_offset,
partition->sb_size,
partition->part_offset+partition->sb_offset,
NULL);
if(partition->upart_type==UP_NTFS ||
(is_part_ntfs(partition) && partition->upart_type!=UP_EXFAT))
ntfs_undelete_part(disk_car, partition, verbose, current_cmd);
else
dir_partition(disk_car, partition, 0, current_cmd);
io_redir_del_redir(disk_car, partition->part_offset+partition->sborg_offset);
}
else
{
if(partition->upart_type==UP_NTFS ||
(is_part_ntfs(partition) && partition->upart_type!=UP_EXFAT))
ntfs_undelete_part(disk_car, partition, verbose, current_cmd);
else
dir_partition(disk_car, partition, 0, current_cmd);
}
}
rewrite=1;
break;
case 'l':
case 'L':
{
partition_t *partition=current_element->part;
if(partition->sb_offset!=0 && partition->sb_size>0)
{
io_redir_add_redir(disk_car,
partition->part_offset+partition->sborg_offset,
partition->sb_size,
partition->part_offset+partition->sb_offset,
NULL);
dir_partition(disk_car,partition,verbose, current_cmd);
io_redir_del_redir(disk_car, partition->part_offset+partition->sborg_offset);
}
else
dir_partition(disk_car,partition,verbose, current_cmd);
}
rewrite=1;
break;
case 's':
case 'S':
{
if(is_linux(current_element->part))
{
list_part_t *list_sb=search_superblock(disk_car,current_element->part,verbose,dump_ind,1);
interface_superblock(disk_car,list_sb,current_cmd);
part_free_list(list_sb);
}
if(is_hfs(current_element->part) || is_hfsp(current_element->part))
{
HFS_HFSP_boot_sector(disk_car, current_element->part, verbose, current_cmd);
}
rewrite=1;
}
break;
case 't':
case 'T':
if(*current_cmd!=NULL)
change_part_type_cli(disk_car, current_element->part, current_cmd);
else
change_part_type_ncurses(disk_car, current_element->part);
rewrite=1;
break;
}
if(current_element_num<offset)
offset=current_element_num;
if(current_element_num>=offset+INTER_ADV)
offset=current_element_num-INTER_ADV+1;
}
#endif
}
}
