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

Update dependencies and optimize download speeds

Browse source
This commit is contained in:
Xpl0itU 2023-12-21 17:02:02 +01:00
parent e7a0240508
commit 9fd0ce9695
6 changed files with 123 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

12
certificate.go
View file

@ -2,11 +2,11 @@ package wiiudownloader
import (
"bytes"
"context"
"fmt"
"net/http"
"os"
"path"
"github.com/cavaliergopher/grab/v3"
)
var cetkData []byte
@ -28,12 +28,12 @@ func getCert(tmdData []byte, id int, numContents uint16) ([]byte, error) {
}
}
func getDefaultCert(progressReporter ProgressReporter, client *grab.Client) ([]byte, error) {
func getDefaultCert(cancelCtx context.Context, progressReporter ProgressReporter, client *http.Client) ([]byte, error) {
if len(cetkData) >= 0x350+0x300 {
return cetkData[0x350 : 0x350+0x300], nil
}
cetkDir := path.Join(os.TempDir(), "cetk")
if err := downloadFile(progressReporter, client, "http://ccs.cdn.c.shop.nintendowifi.net/ccs/download/000500101000400a/cetk", cetkDir, true); err != nil {
if err := downloadFile(cancelCtx, progressReporter, client, "http://ccs.cdn.c.shop.nintendowifi.net/ccs/download/000500101000400a/cetk", cetkDir, true); err != nil {
return nil, err
}
cetkData, err := os.ReadFile(cetkDir)
@ -51,7 +51,7 @@ func getDefaultCert(progressReporter ProgressReporter, client *grab.Client) ([]b
return nil, fmt.Errorf("failed to download OSv10 cetk, length: %d", len(cetkData))
}
func GenerateCert(tmdData []byte, contentCount uint16, progressReporter ProgressReporter, client *grab.Client) (bytes.Buffer, error) {
func GenerateCert(tmdData []byte, contentCount uint16, progressReporter ProgressReporter, client *http.Client, cancelCtx context.Context) (bytes.Buffer, error) {
cert := bytes.Buffer{}
cert0, err := getCert(tmdData, 0, contentCount)
@ -66,7 +66,7 @@ func GenerateCert(tmdData []byte, contentCount uint16, progressReporter Progress
}
cert.Write(cert1)
defaultCert, err := getDefaultCert(progressReporter, client)
defaultCert, err := getDefaultCert(cancelCtx, progressReporter, client)
if err != nil {
return bytes.Buffer{}, err
}