From c9924347e9d3331cf0433752b69421dc06a78217 Mon Sep 17 00:00:00 2001 From: Maschell Date: Sun, 20 Dec 2020 15:29:48 +0100 Subject: [PATCH] Add support for RemoteLocal --- .../wiiu/jnus/NUSTitleLoaderRemoteLocal.java | 44 ++++++++ .../NUSDataProviderRemoteLocal.java | 106 ++++++++++++++++++ 2 files changed, 150 insertions(+) create mode 100644 src/main/java/de/mas/wiiu/jnus/NUSTitleLoaderRemoteLocal.java create mode 100644 src/main/java/de/mas/wiiu/jnus/implementations/NUSDataProviderRemoteLocal.java diff --git a/src/main/java/de/mas/wiiu/jnus/NUSTitleLoaderRemoteLocal.java b/src/main/java/de/mas/wiiu/jnus/NUSTitleLoaderRemoteLocal.java new file mode 100644 index 0000000..829bd6b --- /dev/null +++ b/src/main/java/de/mas/wiiu/jnus/NUSTitleLoaderRemoteLocal.java @@ -0,0 +1,44 @@ +/**************************************************************************** + * Copyright (C) 2016-2019 Maschell + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + ****************************************************************************/ + +package de.mas.wiiu.jnus; + +import de.mas.wiiu.jnus.entities.Ticket; +import de.mas.wiiu.jnus.implementations.DefaultNUSDataProcessor; +import de.mas.wiiu.jnus.implementations.NUSDataProviderLocalBackup; +import de.mas.wiiu.jnus.implementations.NUSDataProviderRemoteLocal; + +public final class NUSTitleLoaderRemoteLocal { + + private NUSTitleLoaderRemoteLocal() { + + } + + public static NUSTitle loadNUSTitle(String inputPath, short titleVersion, Ticket ticket) throws Exception { + NUSTitleConfig config = new NUSTitleConfig(); + + if (ticket != null) { + config.setTicket(ticket); + } else { + config.setNoDecryption(true); + } + + return NUSTitleLoader.loadNusTitle(config, () -> new NUSDataProviderRemoteLocal(inputPath, titleVersion), + (dp, cd, en) -> new DefaultNUSDataProcessor(dp, cd)); + } + +} diff --git a/src/main/java/de/mas/wiiu/jnus/implementations/NUSDataProviderRemoteLocal.java b/src/main/java/de/mas/wiiu/jnus/implementations/NUSDataProviderRemoteLocal.java new file mode 100644 index 0000000..2925f83 --- /dev/null +++ b/src/main/java/de/mas/wiiu/jnus/implementations/NUSDataProviderRemoteLocal.java @@ -0,0 +1,106 @@ +/**************************************************************************** + * Copyright (C) 2016-2019 Maschell + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + ****************************************************************************/ +package de.mas.wiiu.jnus.implementations; + +import java.io.File; +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.IOException; +import java.io.InputStream; +import java.nio.file.Files; +import java.util.Optional; + +import de.mas.wiiu.jnus.Settings; +import de.mas.wiiu.jnus.entities.TMD.Content; +import de.mas.wiiu.jnus.interfaces.NUSDataProvider; +import de.mas.wiiu.jnus.utils.StreamUtils; +import lombok.Getter; + +public class NUSDataProviderRemoteLocal implements NUSDataProvider { + @Getter private final String localPath; + private final short titleVersion; + + public NUSDataProviderRemoteLocal(String localPath) { + this(localPath, (short) Settings.LATEST_TMD_VERSION); + } + + public NUSDataProviderRemoteLocal(String localPath, short version) { + this.localPath = localPath; + this.titleVersion = version; + } + + private String getFilePathOnDisk(Content c) { + return getLocalPath() + File.separator + String.format("%08X", c.getID()); + } + + @Override + public InputStream readRawContentAsStream(Content content, long offset, long size) throws IOException { + File filepath = new File(getFilePathOnDisk(content)); + if (!filepath.exists()) { + throw new FileNotFoundException(filepath.getAbsolutePath() + " was not found."); + } + InputStream in = new FileInputStream(filepath); + StreamUtils.skipExactly(in, offset); + return in; + } + + @Override + public Optional getContentH3Hash(Content content) throws IOException { + String h3Path = getLocalPath() + File.separator + String.format("%08X.h3", content.getID()); + File h3File = new File(h3Path); + if (!h3File.exists()) { + throw new FileNotFoundException(h3File.getAbsolutePath() + " was not found."); + } + return Optional.of(Files.readAllBytes(h3File.toPath())); + } + + @Override + public Optional getRawTMD() throws IOException { + String inputPath = getLocalPath(); + String tmdPath = inputPath + File.separator+ "tmd"; + if (titleVersion != Settings.LATEST_TMD_VERSION) { + tmdPath = inputPath + File.separator+ "tmd." + titleVersion; + } + File tmdFile = new File(tmdPath); + return Optional.of(Files.readAllBytes(tmdFile.toPath())); + } + + @Override + public Optional getRawTicket() throws IOException { + String inputPath = getLocalPath(); + String ticketPath = inputPath + File.separator + "cetk"; + File ticketFile = new File(ticketPath); + return Optional.of(Files.readAllBytes(ticketFile.toPath())); + } + + @Override + public Optional getRawCert() throws IOException { + return Optional.empty(); + } + + @Override + public void cleanup() throws IOException { + // We don't need this + } + + @Override + public String toString() { + String titleVersionString = titleVersion == Settings.LATEST_TMD_VERSION ? "latest" : Short.toString(titleVersion); + return "NUSDataProviderLocalBackup [localPath=" + localPath + ", titleVersion=" + titleVersionString + "]"; + } + +}