mirrors/WiiUDownloader
1
0
Fork 0
mirror of https://github.com/Xpl0itU/WiiUDownloader.git synced 2025-05-28 22:16:04 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify tmd parsing

Browse source
This commit is contained in:
Xpl0itU 2024-04-15 16:27:31 +02:00
parent 91d88dc58e
commit fdff675462
2 changed files with 62 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

49
downloader.go
View file

@ -1,7 +1,6 @@
package wiiudownloader
import (
"bytes"
"context"
"encoding/binary"
"fmt"
@ -185,8 +184,8 @@ func DownloadTitle(titleID, outputDirectory string, doDecryption bool, progressR
return err
}
var titleVersion uint16
if err := binary.Read(bytes.NewReader(tmdData[476:478]), binary.BigEndian, &titleVersion); err != nil {
tmd, err := parseTMD(tmdData)
if err != nil {
return err
}
@ -199,44 +198,20 @@ func DownloadTitle(titleID, outputDirectory string, doDecryption bool, progressR
if err != nil {
return err
}
if err := GenerateTicket(tikPath, tEntry.TitleID, titleKey, titleVersion); err != nil {
if err := GenerateTicket(tikPath, tEntry.TitleID, titleKey, tmd.TitleVersion); err != nil {
return err
}
}
var contentCount uint16
if err := binary.Read(bytes.NewReader(tmdData[478:480]), binary.BigEndian, &contentCount); err != nil {
return err
}
var titleSize uint64
contents := make([]Content, contentCount)
tmdDataReader := bytes.NewReader(tmdData)
for i := 0; i < int(contentCount); i++ {
offset := 0xB04 + (0x30 * i)
tmdDataReader.Seek(int64(offset), io.SeekStart)
if err := binary.Read(tmdDataReader, binary.BigEndian, &contents[i].ID); err != nil {
return err
}
tmdDataReader.Seek(2, io.SeekCurrent)
if err := binary.Read(tmdDataReader, binary.BigEndian, &contents[i].Type); err != nil {
return err
}
if err := binary.Read(tmdDataReader, binary.BigEndian, &contents[i].Size); err != nil {
return err
}
titleSize += contents[i].Size
for i := 0; i < int(tmd.ContentCount); i++ {
titleSize += tmd.Contents[i].Size
}
progressReporter.SetDownloadSize(int64(titleSize))
cert, err := GenerateCert(tmdData, contentCount, progressReporter, client)
cert, err := GenerateCert(tmdData, tmd.ContentCount, progressReporter, client)
if err != nil {
if progressReporter.Cancelled() {
return nil
@ -260,20 +235,20 @@ func DownloadTitle(titleID, outputDirectory string, doDecryption bool, progressR
sem := semaphore.NewWeighted(maxConcurrentDownloads)
progressReporter.SetStartTime(time.Now())
for i := 0; i < int(contentCount); i++ {
for i := 0; i < int(tmd.ContentCount); i++ {
i := i
g.Go(func() error {
filePath := filepath.Join(outputDir, fmt.Sprintf("%08X.app", contents[i].ID))
if err := downloadFileWithSemaphore(ctx, progressReporter, client, fmt.Sprintf("%s/%08X", baseURL, contents[i].ID), filePath, true, sem); err != nil {
filePath := filepath.Join(outputDir, fmt.Sprintf("%08X.app", tmd.Contents[i].ID))
if err := downloadFileWithSemaphore(ctx, progressReporter, client, fmt.Sprintf("%s/%08X", baseURL, tmd.Contents[i].ID), filePath, true, sem); err != nil {
if progressReporter.Cancelled() {
return errCancel
}
return err
}
if contents[i].Type&0x2 == 2 { // has a hash
filePath = filepath.Join(outputDir, fmt.Sprintf("%08X.h3", contents[i].ID))
if err := downloadFileWithSemaphore(ctx, progressReporter, client, fmt.Sprintf("%s/%08X.h3", baseURL, contents[i].ID), filePath, true, sem); err != nil {
if tmd.Contents[i].Type&0x2 == 2 { // has a hash
filePath = filepath.Join(outputDir, fmt.Sprintf("%08X.h3", tmd.Contents[i].ID))
if err := downloadFileWithSemaphore(ctx, progressReporter, client, fmt.Sprintf("%s/%08X.h3", baseURL, tmd.Contents[i].ID), filePath, true, sem); err != nil {
if progressReporter.Cancelled() {
return errCancel
}

50
tmd.go Normal file
View file

@ -0,0 +1,50 @@
package wiiudownloader
import (
"bytes"
"encoding/binary"
"io"
)
type TMD struct {
TitleVersion uint16
ContentCount uint16
Contents []Content
}
func parseTMD(data []byte) (*TMD, error) {
tmd := &TMD{}
reader := bytes.NewReader(data)
reader.Seek(476, io.SeekStart)
if err := binary.Read(reader, binary.BigEndian, &tmd.TitleVersion); err != nil {
return nil, err
}
if err := binary.Read(reader, binary.BigEndian, &tmd.ContentCount); err != nil {
return nil, err
}
tmd.Contents = make([]Content, tmd.ContentCount)
for i := 0; i < int(tmd.ContentCount); i++ {
offset := 0xB04 + (0x30 * i)
reader.Seek(int64(offset), io.SeekStart)
if err := binary.Read(reader, binary.BigEndian, &tmd.Contents[i].ID); err != nil {
return nil, err
}
reader.Seek(2, io.SeekCurrent)
if err := binary.Read(reader, binary.BigEndian, &tmd.Contents[i].Type); err != nil {
return nil, err
}
if err := binary.Read(reader, binary.BigEndian, &tmd.Contents[i].Size); err != nil {
return nil, err
}
}
return tmd, nil
}