int recover_EXFAT(const disk_t *disk, const struct exfat_super_block *exfat_header, partition_t *partition)
{
if(test_EXFAT(exfat_header)!=0)
return 1;
partition->sborg_offset=0;
partition->sb_size=12 << exfat_header->blocksize_bits;
partition->part_type_i386=P_EXFAT;
partition->part_type_gpt=GPT_ENT_TYPE_MS_BASIC_DATA;
partition->part_size=(uint64_t)le64(exfat_header->nr_sectors) * disk->sector_size;
#ifdef DEBUG_EXFAT
log_info("recover_EXFAT:\n");
log_info("start_sector=%llu\n", (long long unsigned)le64(exfat_header->start_sector));
log_info("blocksize=%u\n", (12<<exfat_header->blocksize_bits));
log_info("part_offset=%llu\n", partition->part_offset);
#endif
if((le64(exfat_header->start_sector) * disk ->sector_size +
(12 << exfat_header->blocksize_bits) == partition->part_offset) ||
(disk->arch==&arch_none && (12 << exfat_header->blocksize_bits) == partition->part_offset))
{
partition->sb_offset=12 << exfat_header->blocksize_bits;
partition->part_offset-=partition->sb_offset;
}
set_EXFAT_info(partition, exfat_header);
return 0;
}
int check_EXFAT(disk_t *disk, partition_t *partition)
{
unsigned char *buffer=(unsigned char*)MALLOC(EXFAT_BS_SIZE);
if(disk->pread(disk, buffer, EXFAT_BS_SIZE, partition->part_offset) != EXFAT_BS_SIZE)
{
free(buffer);
return 1;
}
if(test_EXFAT((struct exfat_super_block*)buffer)!=0)
{
free(buffer);
return 1;
}
set_EXFAT_info(partition, (struct exfat_super_block*)buffer);
free(buffer);
return 0;
}
int exFAT_boot_sector(disk_t *disk, partition_t *partition, const int verbose, char **current_cmd)
{
unsigned char *buffer_bs;
unsigned char *buffer_backup_bs;
const char *options="";
int rescan=1;
#ifdef HAVE_NCURSES
struct MenuItem menu_exFAT[]=
{
{ 'P', "Previous",""},
{ 'N', "Next","" },
{ 'Q', "Quit","Return to Advanced menu"},
{ 'O', "Org. BS","Copy superblock over backup sector"},
{ 'B', "Backup BS","Copy backup superblock over superblock"},
{ 'D', "Dump","Dump superblock and backup superblock"},
{ 0, NULL, NULL }
};
#endif
buffer_bs=(unsigned char*)MALLOC(12 * disk->sector_size);
buffer_backup_bs=(unsigned char*)MALLOC(12 * disk->sector_size);
while(1)
{
#ifdef HAVE_NCURSES
unsigned int menu=0;
#endif
int command;
screen_buffer_reset();
if(rescan==1)
{
int opt_over=0;
int opt_B=0;
int opt_O=0;
options="D";
#ifdef HAVE_NCURSES
aff_copy(stdscr);
wmove(stdscr,4,0);
wprintw(stdscr,"%s",disk->description(disk));
mvwaddstr(stdscr,5,0,msg_PART_HEADER_LONG);
wmove(stdscr,6,0);
aff_part(stdscr,AFF_PART_ORDER|AFF_PART_STATUS,disk,partition);
#endif
log_info("\nexFAT_boot_sector\n");
log_partition(disk,partition);
screen_buffer_add("Boot sector\n");
if(disk->pread(disk, buffer_bs, 12 * disk->sector_size, partition->part_offset) != 12 * disk->sector_size)
{
screen_buffer_add("Bad: can't read exFAT boot record.\n");
memset(buffer_bs,0,12 * disk->sector_size);
}
else if(test_EXFAT((const struct exfat_super_block*)buffer_bs, partition)==0)
{
screen_buffer_add("exFAT OK\n");
opt_O=1;
opt_over=1;
}
else
screen_buffer_add("Bad\n");
screen_buffer_add("\nBackup boot record\n");
if(disk->pread(disk, buffer_backup_bs, 12 * disk->sector_size, partition->part_offset + 12 * disk->sector_size) != 12 * disk->sector_size)
{
screen_buffer_add("Bad: can't read exFAT backup boot record.\n");
memset(buffer_backup_bs,0,12 * disk->sector_size);
}
else if(test_EXFAT((const struct exfat_super_block*)buffer_backup_bs, partition)==0)
{
screen_buffer_add("exFAT OK\n");
opt_B=1;
opt_over=1;
}
else
screen_buffer_add("Bad\n");
screen_buffer_add("\n");
if(memcmp(buffer_bs,buffer_backup_bs,12 * disk->sector_size)==0)
{
screen_buffer_add("Sectors are identical.\n");
opt_over=0;
}
else
{
screen_buffer_add("Sectors are not identical.\n");
}
if(opt_over!=0)
{
if(opt_B!=0 && opt_O!=0)
options="DOB";
else if(opt_B!=0)
options="DB";
else if(opt_O!=0)
options="DO";
}
rescan=0;
}
screen_buffer_to_log();
if(*current_cmd!=NULL)
{
command=0;
while(*current_cmd[0]==',')
(*current_cmd)++;
if(strncmp(*current_cmd,"dump",4)==0)
{
(*current_cmd)+=4;
command='D';
}
else if(strncmp(*current_cmd,"originalexFAT",11)==0)
{
(*current_cmd)+=11;
if(strchr(options,'O')!=NULL)
command='O';
}
else if(strncmp(*current_cmd,"backupexFAT",9)==0)
{
(*current_cmd)+=9;
if(strchr(options,'B')!=NULL)
command='B';
}
}
else
{
log_flush();
#ifdef HAVE_NCURSES
command=screen_buffer_display_ext(stdscr, options, menu_exFAT, &menu);
#else
command=0;
#endif
}
switch(command)
{
case 0:
free(buffer_bs);
free(buffer_backup_bs);
return 0;
case 'O': /* O : copy original superblock over backup boot */
#ifdef HAVE_NCURSES
if(ask_confirmation("Copy original exFAT boot record over backup, confirm ? (Y/N)")!=0)
{
log_info("copy original superblock over backup boot\n");
if(disk->pwrite(disk, buffer_bs, 12 * disk->sector_size, partition->part_offset + 12 * disk->sector_size) != 12 * disk->sector_size)
{
display_message("Write error: Can't overwrite exFAT backup boot record\n");
}
disk->sync(disk);
rescan=1;
}
#endif
break;
case 'B': /* B : copy backup superblock over main superblock */
#ifdef HAVE_NCURSES
if(ask_confirmation("Copy backup exFAT boot record over main boot record, confirm ? (Y/N)")!=0)
{
log_info("copy backup superblock over main superblock\n");
/* Reset information about backup boot record */
partition->sb_offset=0;
if(disk->pwrite(disk, buffer_backup_bs, 12 * disk->sector_size, partition->part_offset) != 12 * disk->sector_size)
{
display_message("Write error: Can't overwrite exFAT main boot record\n");
}
disk->sync(disk);
rescan=1;
}
#endif
break;
case 'D':
exFAT_dump(disk, partition, buffer_bs, buffer_backup_bs, current_cmd);
break;
}
}
}
