Add support for Wii's certificate generation

This commit is contained in:
Xpl0itU 2024-04-16 15:34:18 +02:00
parent a6ccbb2473
commit e93a238be9
5 changed files with 29 additions and 46 deletions

View file

@ -10,23 +10,6 @@ import (
var cetkData []byte
func getCert(tmdData []byte, id int, numContents uint16) ([]byte, error) { // TODO: Add support for Wii's TMD
var certSlice []byte
if len(tmdData) == int((0x0B04+0x30*numContents+0xA00)-0x300) {
certSlice = tmdData[0x0B04+0x30*numContents : 0x0B04+0x30*numContents+0xA00-0x300]
} else {
certSlice = tmdData[0x0B04+0x30*numContents : 0x0B04+0x30*numContents+0xA00]
}
switch id {
case 0:
return certSlice[:0x400], nil
case 1:
return certSlice[0x400 : 0x400+0x300], nil
default:
return nil, fmt.Errorf("invalid id: %d", id)
}
}
func getDefaultCert(progressReporter ProgressReporter, client *http.Client) ([]byte, error) {
if len(cetkData) >= 0x350+0x300 {
return cetkData[0x350 : 0x350+0x300], nil
@ -50,20 +33,16 @@ func getDefaultCert(progressReporter ProgressReporter, client *http.Client) ([]b
return nil, fmt.Errorf("failed to download OSv10 cetk, length: %d", len(cetkData))
}
func GenerateCert(tmdData []byte, contentCount uint16, progressReporter ProgressReporter, client *http.Client) (bytes.Buffer, error) {
func GenerateCert(tmd *TMD, progressReporter ProgressReporter, client *http.Client) (bytes.Buffer, error) {
cert := bytes.Buffer{}
cert0, err := getCert(tmdData, 0, contentCount)
if err != nil {
if _, err := cert.Write(tmd.Certificate1); err != nil {
return bytes.Buffer{}, err
}
cert.Write(cert0)
cert1, err := getCert(tmdData, 1, contentCount)
if err != nil {
if _, err := cert.Write(tmd.Certificate2); err != nil {
return bytes.Buffer{}, err
}
cert.Write(cert1)
defaultCert, err := getDefaultCert(progressReporter, client)
if err != nil {