int wbfs_read_wii_file(void *_handle, u32 _offset, u32 count, void *buf) { HANDLE *handle = (HANDLE *)_handle; LARGE_INTEGER large; DWORD read; u64 offset = _offset; offset <<= 2; large.QuadPart = offset; if (SetFilePointerEx(handle, large, NULL, FILE_BEGIN) == FALSE) { wbfs_error("error seeking in disc file"); return 1; } read = 0; if ((ReadFile(handle, buf, count, &read, NULL) == FALSE) || !read) { wbfs_error("error reading wii disc sector"); return 1; } if (read < count) { wbfs_warning("warning: requested %d, but read only %d bytes (trimmed or bad padded ISO)", count, read); wbfs_memset((u8*)buf+read, 0, count-read); } return 0; }
static void _decrypt_title_key(u8 *tik, u8 *title_key) { u8 common_key[16] = { 0xeb, 0xe4, 0x2a, 0x22, 0x5e, 0x85, 0x93, 0xe4, 0x48, 0xd9, 0xc5, 0x45, 0x73, 0x81, 0xaa, 0xf7 }; u8 iv[16]; wbfs_memset(iv, 0, sizeof iv); wbfs_memcpy(iv, tik + 0x01dc, 8); aes_set_key(common_key); //_aes_cbc_dec(common_key, iv, tik + 0x01bf, 16, title_key); aes_decrypt(iv, tik + 0x01bf, title_key, 16); }
wbfs_t*wbfs_open_partition(rw_sector_callback_t read_hdsector, rw_sector_callback_t write_hdsector, #ifdef WIN32 close_callback_t close_hd, #endif 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; #ifdef WIN32 p->close_hd = close_hd; #endif 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; wbfs_sync(p); return p; error: wbfs_free(p); wbfs_iofree(head); return 0; }