Add support for RemoteLocal

This commit is contained in:
Maschell 2020-12-20 15:29:48 +01:00
parent 728e1f01bd
commit c9924347e9
2 changed files with 150 additions and 0 deletions

View File

@ -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 <http://www.gnu.org/licenses/>.
****************************************************************************/
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));
}
}

View File

@ -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 <http://www.gnu.org/licenses/>.
****************************************************************************/
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<byte[]> 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<byte[]> 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<byte[]> 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<byte[]> 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 + "]";
}
}