mirror of
https://github.com/wiidev/usbloadergx.git
synced 2024-12-23 10:21:55 +01:00
a09abe355f
* Fixed issue 1058 * Menus splitted to several smaller files, to reduce compile time This version has FAT support. You can change the used partition in the game load options. Default WBFS will be used, if found. Otherwise the first FAT partition with games will be used. FAT will only work when using Hermes cios (222/223)!!!
694 lines
23 KiB
C
694 lines
23 KiB
C
// Copyright 2009 Kwiirk
|
|
// Licensed under the terms of the GNU GPL, version 2
|
|
// http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt
|
|
|
|
|
|
#include "libwbfs.h"
|
|
|
|
#define likely(x) __builtin_expect(!!(x), 1)
|
|
#define unlikely(x) __builtin_expect(!!(x), 0)
|
|
|
|
#define ERROR(x) do {wbfs_error(x);goto error;}while(0)
|
|
#define ALIGN_LBA(x) (((x)+p->hd_sec_sz-1)&(~(p->hd_sec_sz-1)))
|
|
static int force_mode=0;
|
|
void wbfs_set_force_mode(int force)
|
|
{
|
|
force_mode = force;
|
|
}
|
|
static u8 size_to_shift(u32 size)
|
|
{
|
|
u8 ret = 0;
|
|
while(size)
|
|
{
|
|
ret++;
|
|
size>>=1;
|
|
}
|
|
return ret-1;
|
|
}
|
|
#define read_le32_unaligned(x) ((x)[0]|((x)[1]<<8)|((x)[2]<<16)|((x)[3]<<24))
|
|
|
|
|
|
wbfs_t*wbfs_open_hd(rw_sector_callback_t read_hdsector,
|
|
rw_sector_callback_t write_hdsector,
|
|
void *callback_data,
|
|
int hd_sector_size, int num_hd_sector __attribute((unused)), int reset)
|
|
{
|
|
int i=num_hd_sector,ret;
|
|
u8 *ptr,*tmp_buffer = wbfs_ioalloc(hd_sector_size);
|
|
u8 part_table[16*4];
|
|
ret = read_hdsector(callback_data,0,1,tmp_buffer);
|
|
if(ret)
|
|
return 0;
|
|
//find wbfs partition
|
|
wbfs_memcpy(part_table,tmp_buffer+0x1be,16*4);
|
|
ptr = part_table;
|
|
for(i=0;i<4;i++,ptr+=16)
|
|
{
|
|
u32 part_lba = read_le32_unaligned(ptr+0x8);
|
|
wbfs_head_t *head = (wbfs_head_t *)tmp_buffer;
|
|
ret = read_hdsector(callback_data,part_lba,1,tmp_buffer);
|
|
// verify there is the magic.
|
|
if (head->magic == wbfs_htonl(WBFS_MAGIC))
|
|
{
|
|
// Override the sector size by the sector size in the wbfs header...
|
|
hd_sector_size = 1 << head->hd_sec_sz_s;
|
|
|
|
wbfs_t*p = wbfs_open_partition(read_hdsector,write_hdsector,
|
|
callback_data,hd_sector_size,0,part_lba,reset);
|
|
return p;
|
|
}
|
|
}
|
|
if(reset)// XXX make a empty hd partition..
|
|
{
|
|
}
|
|
return 0;
|
|
}
|
|
wbfs_t*wbfs_open_partition(rw_sector_callback_t read_hdsector,
|
|
rw_sector_callback_t write_hdsector,
|
|
void *callback_data,
|
|
int hd_sector_size, int num_hd_sector, u32 part_lba, int reset)
|
|
{
|
|
wbfs_t *p = wbfs_malloc(sizeof(wbfs_t));
|
|
|
|
wbfs_head_t *head = wbfs_ioalloc(hd_sector_size?hd_sector_size:512);
|
|
|
|
//constants, but put here for consistancy
|
|
p->wii_sec_sz = 0x8000;
|
|
p->wii_sec_sz_s = size_to_shift(0x8000);
|
|
p->n_wii_sec = (num_hd_sector/0x8000)*hd_sector_size;
|
|
p->n_wii_sec_per_disc = 143432*2;//support for double layers discs..
|
|
p->head = head;
|
|
p->part_lba = part_lba;
|
|
// init the partition
|
|
if (reset)
|
|
{
|
|
u8 sz_s;
|
|
wbfs_memset(head,0,hd_sector_size);
|
|
head->magic = wbfs_htonl(WBFS_MAGIC);
|
|
head->hd_sec_sz_s = size_to_shift(hd_sector_size);
|
|
head->n_hd_sec = wbfs_htonl(num_hd_sector);
|
|
// choose minimum wblk_sz that fits this partition size
|
|
for(sz_s=6;sz_s<11;sz_s++)
|
|
{
|
|
// ensure that wbfs_sec_sz is big enough to address every blocks using 16 bits
|
|
if(p->n_wii_sec <((1U<<16)*(1<<sz_s)))
|
|
break;
|
|
}
|
|
head->wbfs_sec_sz_s = sz_s+p->wii_sec_sz_s;
|
|
}else
|
|
read_hdsector(callback_data,p->part_lba,1,head);
|
|
if (head->magic != wbfs_htonl(WBFS_MAGIC))
|
|
ERROR("bad magic");
|
|
if(!force_mode && hd_sector_size && head->hd_sec_sz_s != size_to_shift(hd_sector_size))
|
|
ERROR("hd sector size doesn't match");
|
|
if(!force_mode && num_hd_sector && head->n_hd_sec != wbfs_htonl(num_hd_sector))
|
|
ERROR("hd num sector doesn't match");
|
|
p->hd_sec_sz = 1<<head->hd_sec_sz_s;
|
|
p->hd_sec_sz_s = head->hd_sec_sz_s;
|
|
p->n_hd_sec = wbfs_ntohl(head->n_hd_sec);
|
|
|
|
p->n_wii_sec = (p->n_hd_sec/p->wii_sec_sz)*(p->hd_sec_sz);
|
|
|
|
p->wbfs_sec_sz_s = head->wbfs_sec_sz_s;
|
|
p->wbfs_sec_sz = 1<<p->wbfs_sec_sz_s;
|
|
p->n_wbfs_sec = p->n_wii_sec >> (p->wbfs_sec_sz_s - p->wii_sec_sz_s);
|
|
p->n_wbfs_sec_per_disc = p->n_wii_sec_per_disc >> (p->wbfs_sec_sz_s - p->wii_sec_sz_s);
|
|
p->disc_info_sz = ALIGN_LBA(sizeof(wbfs_disc_info_t) + p->n_wbfs_sec_per_disc*2);
|
|
|
|
//printf("hd_sector_size %X wii_sector size %X wbfs sector_size %X\n",p->hd_sec_sz,p->wii_sec_sz,p->wbfs_sec_sz);
|
|
p->read_hdsector = read_hdsector;
|
|
p->write_hdsector = write_hdsector;
|
|
p->callback_data = callback_data;
|
|
|
|
p->freeblks_lba = (p->wbfs_sec_sz - p->n_wbfs_sec/8)>>p->hd_sec_sz_s;
|
|
|
|
if(!reset)
|
|
p->freeblks = 0; // will alloc and read only if needed
|
|
else
|
|
{
|
|
// init with all free blocks
|
|
p->freeblks = wbfs_ioalloc(ALIGN_LBA(p->n_wbfs_sec/8));
|
|
wbfs_memset(p->freeblks,0xff,p->n_wbfs_sec/8);
|
|
}
|
|
p->max_disc = (p->freeblks_lba-1)/(p->disc_info_sz>>p->hd_sec_sz_s);
|
|
if(p->max_disc > p->hd_sec_sz - sizeof(wbfs_head_t))
|
|
p->max_disc = p->hd_sec_sz - sizeof(wbfs_head_t);
|
|
|
|
p->tmp_buffer = wbfs_ioalloc(p->hd_sec_sz);
|
|
p->n_disc_open = 0;
|
|
return p;
|
|
error:
|
|
wbfs_free(p);
|
|
wbfs_iofree(head);
|
|
return 0;
|
|
|
|
}
|
|
|
|
void wbfs_sync(wbfs_t*p)
|
|
{
|
|
// copy back descriptors
|
|
if(p->write_hdsector){
|
|
p->write_hdsector(p->callback_data,p->part_lba+0,1, p->head);
|
|
|
|
if(p->freeblks)
|
|
p->write_hdsector(p->callback_data,p->part_lba+p->freeblks_lba,ALIGN_LBA(p->n_wbfs_sec/8)>>p->hd_sec_sz_s, p->freeblks);
|
|
}
|
|
}
|
|
void wbfs_close(wbfs_t*p)
|
|
{
|
|
wbfs_sync(p);
|
|
|
|
if(p->n_disc_open)
|
|
ERROR("trying to close wbfs while discs still open");
|
|
|
|
wbfs_iofree(p->head);
|
|
wbfs_iofree(p->tmp_buffer);
|
|
if(p->freeblks)
|
|
wbfs_iofree(p->freeblks);
|
|
|
|
wbfs_free(p);
|
|
|
|
error:
|
|
return;
|
|
}
|
|
|
|
wbfs_disc_t *wbfs_open_disc(wbfs_t* p, u8 *discid)
|
|
{
|
|
u32 i;
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
wbfs_disc_t *d = 0;
|
|
for(i=0;i<p->max_disc;i++)
|
|
{
|
|
if (p->head->disc_table[i]){
|
|
p->read_hdsector(p->callback_data,
|
|
p->part_lba+1+i*disc_info_sz_lba,1,p->tmp_buffer);
|
|
if(wbfs_memcmp(discid,p->tmp_buffer,6)==0){
|
|
d = wbfs_malloc(sizeof(*d));
|
|
if(!d)
|
|
ERROR("allocating memory");
|
|
d->p = p;
|
|
d->i = i;
|
|
d->header = wbfs_ioalloc(p->disc_info_sz);
|
|
if(!d->header)
|
|
ERROR("allocating memory");
|
|
p->read_hdsector(p->callback_data,
|
|
p->part_lba+1+i*disc_info_sz_lba,
|
|
disc_info_sz_lba,d->header);
|
|
p->n_disc_open ++;
|
|
// for(i=0;i<p->n_wbfs_sec_per_disc;i++)
|
|
// printf("%d,",wbfs_ntohs(d->header->wlba_table[i]));
|
|
return d;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
error:
|
|
if(d)
|
|
wbfs_iofree(d);
|
|
return 0;
|
|
|
|
}
|
|
void wbfs_close_disc(wbfs_disc_t*d)
|
|
{
|
|
d->p->n_disc_open --;
|
|
wbfs_iofree(d->header);
|
|
wbfs_free(d);
|
|
}
|
|
// offset is pointing 32bit words to address the whole dvd, although len is in bytes
|
|
int wbfs_disc_read(wbfs_disc_t*d,u32 offset, u32 len, u8 *data)
|
|
{
|
|
|
|
wbfs_t *p = d->p;
|
|
u16 wlba = offset>>(p->wbfs_sec_sz_s-2);
|
|
u32 iwlba_shift = p->wbfs_sec_sz_s - p->hd_sec_sz_s;
|
|
u32 lba_mask = (p->wbfs_sec_sz-1)>>(p->hd_sec_sz_s);
|
|
u32 lba = (offset>>(p->hd_sec_sz_s-2))&lba_mask;
|
|
u32 off = offset&((p->hd_sec_sz>>2)-1);
|
|
u16 iwlba = wbfs_ntohs(d->header->wlba_table[wlba]);
|
|
u32 len_copied;
|
|
int err = 0;
|
|
u8 *ptr = data;
|
|
if(unlikely(iwlba==0))
|
|
return 1;
|
|
if(unlikely(off)){
|
|
off*=4;
|
|
err = p->read_hdsector(p->callback_data,
|
|
p->part_lba + (iwlba<<iwlba_shift) + lba, 1, p->tmp_buffer);
|
|
if(err)
|
|
return err;
|
|
len_copied = p->hd_sec_sz - off;
|
|
if(likely(len < len_copied))
|
|
len_copied = len;
|
|
wbfs_memcpy(ptr, p->tmp_buffer + off, len_copied);
|
|
len -= len_copied;
|
|
ptr += len_copied;
|
|
lba++;
|
|
if(unlikely(lba>lba_mask && len)){
|
|
lba=0;
|
|
iwlba = wbfs_ntohs(d->header->wlba_table[++wlba]);
|
|
if(unlikely(iwlba==0))
|
|
return 1;
|
|
}
|
|
}
|
|
while(likely(len>=p->hd_sec_sz))
|
|
{
|
|
u32 nlb = len>>(p->hd_sec_sz_s);
|
|
|
|
if(unlikely(lba + nlb > p->wbfs_sec_sz)) // dont cross wbfs sectors..
|
|
nlb = p->wbfs_sec_sz-lba;
|
|
err = p->read_hdsector(p->callback_data,
|
|
p->part_lba + (iwlba<<iwlba_shift) + lba, nlb, ptr);
|
|
if(err)
|
|
return err;
|
|
len -= nlb<<p->hd_sec_sz_s;
|
|
ptr += nlb<<p->hd_sec_sz_s;
|
|
lba += nlb;
|
|
if(unlikely(lba>lba_mask && len)){
|
|
lba = 0;
|
|
iwlba =wbfs_ntohs(d->header->wlba_table[++wlba]);
|
|
if(unlikely(iwlba==0))
|
|
return 1;
|
|
}
|
|
}
|
|
if(unlikely(len)){
|
|
err = p->read_hdsector(p->callback_data,
|
|
p->part_lba + (iwlba<<iwlba_shift) + lba, 1, p->tmp_buffer);
|
|
if(err)
|
|
return err;
|
|
wbfs_memcpy(ptr, p->tmp_buffer, len);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// disc listing
|
|
u32 wbfs_count_discs(wbfs_t*p)
|
|
{
|
|
u32 i,count=0;
|
|
for(i=0;i<p->max_disc;i++)
|
|
if (p->head->disc_table[i])
|
|
count++;
|
|
return count;
|
|
|
|
}
|
|
u32 wbfs_sector_used(wbfs_t *p,wbfs_disc_info_t *di)
|
|
{
|
|
u32 tot_blk=0,j;
|
|
for(j=0;j<p->n_wbfs_sec_per_disc;j++)
|
|
if(wbfs_ntohs(di->wlba_table[j]))
|
|
tot_blk++;
|
|
return tot_blk;
|
|
|
|
}
|
|
u32 wbfs_get_disc_info(wbfs_t*p, u32 index,u8 *header,int header_size,u32 *size)//size in 32 bit
|
|
{
|
|
u32 i,count=0;
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
for(i=0;i<p->max_disc;i++)
|
|
if (p->head->disc_table[i]){
|
|
if(count++==index)
|
|
{
|
|
p->read_hdsector(p->callback_data,
|
|
p->part_lba+1+i*disc_info_sz_lba,1,p->tmp_buffer);
|
|
if(header_size > (int)p->hd_sec_sz)
|
|
header_size = p->hd_sec_sz;
|
|
u32 magic = wbfs_ntohl(*(u32*)(p->tmp_buffer+24));
|
|
if(magic!=0x5D1C9EA3){
|
|
p->head->disc_table[i]=0;
|
|
return 1;
|
|
}
|
|
memcpy(header,p->tmp_buffer,header_size);
|
|
if(size)
|
|
{
|
|
u8 *header = wbfs_ioalloc(p->disc_info_sz);
|
|
p->read_hdsector(p->callback_data,
|
|
p->part_lba+1+i*disc_info_sz_lba,disc_info_sz_lba,header);
|
|
u32 sec_used = wbfs_sector_used(p,(wbfs_disc_info_t *)header);
|
|
wbfs_iofree(header);
|
|
*size = sec_used<<(p->wbfs_sec_sz_s-2);
|
|
}
|
|
return 0;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static void load_freeblocks(wbfs_t*p)
|
|
{
|
|
if(p->freeblks)
|
|
return;
|
|
// XXX should handle malloc error..
|
|
p->freeblks = wbfs_ioalloc(ALIGN_LBA(p->n_wbfs_sec/8));
|
|
p->read_hdsector(p->callback_data,p->part_lba+p->freeblks_lba,ALIGN_LBA(p->n_wbfs_sec/8)>>p->hd_sec_sz_s, p->freeblks);
|
|
|
|
}
|
|
u32 wbfs_count_usedblocks(wbfs_t*p)
|
|
{
|
|
u32 i,j,count=0;
|
|
load_freeblocks(p);
|
|
for(i=0;i<p->n_wbfs_sec/(8*4);i++)
|
|
{
|
|
u32 v = wbfs_ntohl(p->freeblks[i]);
|
|
if(v == ~0U)
|
|
count+=32;
|
|
else if(v!=0)
|
|
for(j=0;j<32;j++)
|
|
if (v & (1<<j))
|
|
count++;
|
|
}
|
|
return count;
|
|
}
|
|
|
|
|
|
// write access
|
|
|
|
|
|
static int block_used(u8 *used,u32 i,u32 wblk_sz)
|
|
{
|
|
u32 k;
|
|
i*=wblk_sz;
|
|
for(k=0;k<wblk_sz;k++)
|
|
if(i+k<143432*2 && used[i+k])
|
|
return 1;
|
|
return 0;
|
|
}
|
|
|
|
static u32 alloc_block(wbfs_t*p)
|
|
{
|
|
u32 i,j;
|
|
for(i=0;i<p->n_wbfs_sec/(8*4);i++)
|
|
{
|
|
u32 v = wbfs_ntohl(p->freeblks[i]);
|
|
if(v != 0)
|
|
{
|
|
for(j=0;j<32;j++)
|
|
if (v & (1<<j))
|
|
{
|
|
p->freeblks[i] = wbfs_htonl(v & ~(1<<j));
|
|
return (i*32)+j+1;
|
|
}
|
|
}
|
|
}
|
|
return ~0;
|
|
}
|
|
static void free_block(wbfs_t *p,int bl)
|
|
{
|
|
int i = (bl-1)/(32);
|
|
int j = (bl-1)&31;
|
|
u32 v = wbfs_ntohl(p->freeblks[i]);
|
|
p->freeblks[i] = wbfs_htonl(v | 1<<j);
|
|
}
|
|
u32 wbfs_add_disc(wbfs_t*p,read_wiidisc_callback_t read_src_wii_disc,
|
|
void *callback_data,progress_callback_t spinner,partition_selector_t sel,int copy_1_1)
|
|
{
|
|
int i,discn;
|
|
u32 tot,cur;
|
|
u32 wii_sec_per_wbfs_sect = 1<<(p->wbfs_sec_sz_s-p->wii_sec_sz_s);
|
|
wiidisc_t *d = 0;
|
|
u8 *used = 0;
|
|
wbfs_disc_info_t *info = 0;
|
|
u8* copy_buffer = 0;
|
|
used = wbfs_malloc(p->n_wii_sec_per_disc);
|
|
if(!used)
|
|
ERROR("unable to alloc memory");
|
|
if(!copy_1_1)
|
|
{
|
|
d = wd_open_disc(read_src_wii_disc,callback_data);
|
|
if(!d)
|
|
ERROR("unable to open wii disc");
|
|
wd_build_disc_usage(d,sel,used);
|
|
wd_close_disc(d);
|
|
d = 0;
|
|
}
|
|
|
|
|
|
for(i=0;i<p->max_disc;i++)// find a free slot.
|
|
if(p->head->disc_table[i]==0)
|
|
break;
|
|
if(i==p->max_disc)
|
|
ERROR("no space left on device (table full)");
|
|
p->head->disc_table[i] = 1;
|
|
discn = i;
|
|
load_freeblocks(p);
|
|
|
|
// build disc info
|
|
info = wbfs_ioalloc(p->disc_info_sz);
|
|
read_src_wii_disc(callback_data,0,0x100,info->disc_header_copy);
|
|
|
|
copy_buffer = wbfs_ioalloc(p->wii_sec_sz);
|
|
if(!copy_buffer)
|
|
ERROR("alloc memory");
|
|
tot=0;
|
|
cur=0;
|
|
if(spinner){
|
|
// count total number to write for spinner
|
|
for(i=0; i<p->n_wbfs_sec_per_disc;i++)
|
|
if(copy_1_1 || block_used(used,i,wii_sec_per_wbfs_sect)) tot += wii_sec_per_wbfs_sect;
|
|
spinner(0,tot);
|
|
}
|
|
for(i=0; i<p->n_wbfs_sec_per_disc;i++){
|
|
u16 bl = 0;
|
|
if(copy_1_1 || block_used(used,i,wii_sec_per_wbfs_sect)) {
|
|
u16 j;
|
|
|
|
bl = alloc_block(p);
|
|
if (bl==0xffff)
|
|
ERROR("no space left on device (disc full)");
|
|
for(j=0; j<wii_sec_per_wbfs_sect;j++) {
|
|
u32 offset = (i*(p->wbfs_sec_sz>>2)) + (j*(p->wii_sec_sz>>2));
|
|
|
|
read_src_wii_disc(callback_data,offset,p->wii_sec_sz,copy_buffer);
|
|
|
|
//fix the partition table
|
|
if(offset == (0x40000>>2))
|
|
wd_fix_partition_table(d, sel, copy_buffer);
|
|
p->write_hdsector(p->callback_data,p->part_lba+bl*(p->wbfs_sec_sz/p->hd_sec_sz)+j*(p->wii_sec_sz/p->hd_sec_sz),
|
|
p->wii_sec_sz/p->hd_sec_sz,copy_buffer);
|
|
cur++;
|
|
if(spinner)
|
|
spinner(cur,tot);
|
|
}
|
|
}
|
|
info->wlba_table[i] = wbfs_htons(bl);
|
|
}
|
|
// write disc info
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
p->write_hdsector(p->callback_data,p->part_lba+1+discn*disc_info_sz_lba,disc_info_sz_lba,info);
|
|
wbfs_sync(p);
|
|
error:
|
|
if(d)
|
|
wd_close_disc(d);
|
|
if(used)
|
|
wbfs_free(used);
|
|
if(info)
|
|
wbfs_iofree(info);
|
|
if(copy_buffer)
|
|
wbfs_iofree(copy_buffer);
|
|
// init with all free blocks
|
|
|
|
return 0;
|
|
}
|
|
|
|
u32 wbfs_rm_disc(wbfs_t*p, u8* discid)
|
|
{
|
|
wbfs_disc_t *d = wbfs_open_disc(p,discid);
|
|
int i;
|
|
int discn = 0;
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
if(!d)
|
|
return 1;
|
|
load_freeblocks(p);
|
|
discn = d->i;
|
|
for( i=0; i< p->n_wbfs_sec_per_disc; i++)
|
|
{
|
|
u32 iwlba = wbfs_ntohs(d->header->wlba_table[i]);
|
|
if (iwlba)
|
|
free_block(p,iwlba);
|
|
}
|
|
memset(d->header,0,p->disc_info_sz);
|
|
p->write_hdsector(p->callback_data,p->part_lba+1+discn*disc_info_sz_lba,disc_info_sz_lba,d->header);
|
|
p->head->disc_table[discn] = 0;
|
|
wbfs_close_disc(d);
|
|
wbfs_sync(p);
|
|
return 0;
|
|
}
|
|
|
|
u32 wbfs_ren_disc(wbfs_t*p, u8* discid, u8* newname)
|
|
{
|
|
wbfs_disc_t *d = wbfs_open_disc(p,discid);
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
|
|
if(!d)
|
|
return 1;
|
|
|
|
memset(d->header->disc_header_copy+0x20, 0, 0x40);
|
|
strncpy((char *) d->header->disc_header_copy+0x20, (char *) newname, 0x39);
|
|
|
|
p->write_hdsector(p->callback_data,p->part_lba+1+d->i*disc_info_sz_lba,disc_info_sz_lba,d->header);
|
|
wbfs_close_disc(d);
|
|
return 0;
|
|
}
|
|
|
|
u32 wbfs_rID_disc(wbfs_t*p, u8* discid, u8* newID)
|
|
{
|
|
wbfs_disc_t *d = wbfs_open_disc(p,discid);
|
|
int disc_info_sz_lba = p->disc_info_sz>>p->hd_sec_sz_s;
|
|
|
|
if(!d)
|
|
return 1;
|
|
|
|
memset(d->header->disc_header_copy, 0, 0x10);
|
|
strncpy((char *) d->header->disc_header_copy, (char *) newID, 0x9);
|
|
|
|
p->write_hdsector(p->callback_data,p->part_lba+1+d->i*disc_info_sz_lba,disc_info_sz_lba,d->header);
|
|
wbfs_close_disc(d);
|
|
return 0;
|
|
}
|
|
|
|
// trim the file-system to its minimum size
|
|
u32 wbfs_trim(wbfs_t*p);
|
|
|
|
// data extraction
|
|
u32 wbfs_extract_disc(wbfs_disc_t*d, rw_sector_callback_t write_dst_wii_sector,void *callback_data,progress_callback_t spinner)
|
|
{
|
|
wbfs_t *p = d->p;
|
|
u8* copy_buffer = 0;
|
|
int i;
|
|
int src_wbs_nlb=p->wbfs_sec_sz/p->hd_sec_sz;
|
|
int dst_wbs_nlb=p->wbfs_sec_sz/p->wii_sec_sz;
|
|
copy_buffer = wbfs_ioalloc(p->wbfs_sec_sz);
|
|
if(!copy_buffer)
|
|
ERROR("alloc memory");
|
|
|
|
for( i=0; i< p->n_wbfs_sec_per_disc; i++)
|
|
{
|
|
u32 iwlba = wbfs_ntohs(d->header->wlba_table[i]);
|
|
if (iwlba)
|
|
{
|
|
|
|
if(spinner)
|
|
spinner(i,p->n_wbfs_sec_per_disc);
|
|
p->read_hdsector(p->callback_data, p->part_lba + iwlba*src_wbs_nlb, src_wbs_nlb, copy_buffer);
|
|
write_dst_wii_sector(callback_data, i*dst_wbs_nlb, dst_wbs_nlb, copy_buffer);
|
|
}
|
|
}
|
|
wbfs_iofree(copy_buffer);
|
|
return 0;
|
|
error:
|
|
return 1;
|
|
}
|
|
u32 wbfs_extract_file(wbfs_disc_t*d, char *path);
|
|
|
|
float wbfs_estimate_disc
|
|
(
|
|
wbfs_t *p, read_wiidisc_callback_t read_src_wii_disc,
|
|
void *callback_data,
|
|
partition_selector_t sel)
|
|
{
|
|
u8 *b;
|
|
int i;
|
|
u32 tot;
|
|
u32 wii_sec_per_wbfs_sect = 1 << (p->wbfs_sec_sz_s-p->wii_sec_sz_s);
|
|
wiidisc_t *d = 0;
|
|
u8 *used = 0;
|
|
wbfs_disc_info_t *info = 0;
|
|
|
|
tot = 0;
|
|
|
|
used = wbfs_malloc(p->n_wii_sec_per_disc);
|
|
if (!used)
|
|
{
|
|
ERROR("unable to alloc memory");
|
|
}
|
|
|
|
d = wd_open_disc(read_src_wii_disc, callback_data);
|
|
if (!d)
|
|
{
|
|
ERROR("unable to open wii disc");
|
|
}
|
|
|
|
wd_build_disc_usage(d,sel,used);
|
|
wd_close_disc(d);
|
|
d = 0;
|
|
|
|
info = wbfs_ioalloc(p->disc_info_sz);
|
|
b = (u8 *)info;
|
|
read_src_wii_disc(callback_data, 0, 0x100, info->disc_header_copy);
|
|
|
|
//fprintf(stderr, "estimating %c%c%c%c%c%c %s...\n",b[0], b[1], b[2], b[3], b[4], b[5], b + 0x20);
|
|
|
|
for (i = 0; i < p->n_wbfs_sec_per_disc; i++)
|
|
{
|
|
if (block_used(used, i, wii_sec_per_wbfs_sect))
|
|
{
|
|
tot++;
|
|
}
|
|
}
|
|
//memcpy(header, b,0x100);
|
|
|
|
error:
|
|
if (d)
|
|
wd_close_disc(d);
|
|
|
|
if (used)
|
|
wbfs_free(used);
|
|
|
|
if (info)
|
|
wbfs_iofree(info);
|
|
|
|
return tot * (((p->wbfs_sec_sz*1.0) / p->hd_sec_sz) * 512);
|
|
}
|
|
|
|
u32 wbfs_size_disc(wbfs_t*p,read_wiidisc_callback_t read_src_wii_disc,
|
|
void *callback_data,partition_selector_t sel,
|
|
u32 *comp_size, u32 *real_size)
|
|
{
|
|
int i;
|
|
u32 tot = 0, last = 0;
|
|
u32 wii_sec_per_wbfs_sect = 1<<(p->wbfs_sec_sz_s-p->wii_sec_sz_s);
|
|
wiidisc_t *d = 0;
|
|
u8 *used = 0;
|
|
used = wbfs_malloc(p->n_wii_sec_per_disc);
|
|
if(!used)
|
|
ERROR("unable to alloc memory");
|
|
d = wd_open_disc(read_src_wii_disc,callback_data);
|
|
if(!d)
|
|
ERROR("unable to open wii disc");
|
|
wd_build_disc_usage(d,sel,used);
|
|
wd_close_disc(d);
|
|
d = 0;
|
|
|
|
// count total number to write for spinner
|
|
for(i=0; i<p->n_wbfs_sec_per_disc;i++) {
|
|
if(block_used(used,i,wii_sec_per_wbfs_sect)) {
|
|
tot += wii_sec_per_wbfs_sect;
|
|
last = i * wii_sec_per_wbfs_sect;
|
|
}
|
|
}
|
|
|
|
error:
|
|
if(d)
|
|
wd_close_disc(d);
|
|
if(used)
|
|
wbfs_free(used);
|
|
|
|
*comp_size = tot;
|
|
*real_size = last;
|
|
|
|
return 0;
|
|
}
|
|
|
|
// trim the file-system to its minimum size
|
|
u32 wbfs_trim(wbfs_t*p)
|
|
{
|
|
u32 maxbl;
|
|
load_freeblocks(p);
|
|
maxbl = alloc_block(p);
|
|
p->n_hd_sec = maxbl<<(p->wbfs_sec_sz_s-p->hd_sec_sz_s);
|
|
p->head->n_hd_sec = wbfs_htonl(p->n_hd_sec);
|
|
// make all block full
|
|
memset(p->freeblks,0,p->n_wbfs_sec/8);
|
|
wbfs_sync(p);
|
|
// os layer will truncate the file.
|
|
return maxbl;
|
|
}
|