118 lines
4.6 KiB
Java
118 lines
4.6 KiB
Java
/*
|
|
Copyright 2019-2022 Dmitry Isaenko
|
|
|
|
This file is part of libKonogonka.
|
|
|
|
libKonogonka 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.
|
|
|
|
libKonogonka 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 libKonogonka. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
package libKonogonka.RomFsDecrypted;
|
|
|
|
import libKonogonka.KeyChainHolder;
|
|
import libKonogonka.Tools.other.System2.System2Provider;
|
|
import libKonogonka.Tools.other.System2.ini1.Ini1Provider;
|
|
import libKonogonka.Tools.other.System2.ini1.Kip1;
|
|
import libKonogonka.ctraesclassic.AesCtrDecryptClassic;
|
|
import org.junit.jupiter.api.Assertions;
|
|
import org.junit.jupiter.api.Disabled;
|
|
import org.junit.jupiter.api.DisplayName;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import java.io.*;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
public class Package2UnpackedTest {
|
|
|
|
private static final String keysFileLocation = "./FilesForTests/prod.keys";
|
|
private static final String xci_header_keyFileLocation = "./FilesForTests/xci_header_key.txt";
|
|
private static KeyChainHolder keyChainHolder;
|
|
|
|
private static final String fileLocation = "/home/loper/Projects/libKonogonka/FilesForTests/6b7abe7efa17ad065b18e62d1c87a5cc.nca_extracted/ROOT/nx/package2";
|
|
|
|
@Disabled
|
|
@DisplayName("Package2 unpacked test")
|
|
@Test
|
|
void discover() throws Exception{
|
|
BufferedReader br = new BufferedReader(new FileReader(xci_header_keyFileLocation));
|
|
String keyValue = br.readLine();
|
|
br.close();
|
|
|
|
Assertions.assertNotNull(keyValue);
|
|
|
|
keyValue = keyValue.trim();
|
|
keyChainHolder = new KeyChainHolder(keysFileLocation, keyValue);
|
|
|
|
HashMap<String, String> rawKeys = keyChainHolder.getRawKeySet();
|
|
|
|
HashMap<String, String> package2_keys = new HashMap<>();
|
|
|
|
for (String key: rawKeys.keySet()){
|
|
if (key.matches("package2_key_[0-f][0-f]"))
|
|
package2_keys.put(key, rawKeys.get(key));
|
|
}
|
|
|
|
Assertions.assertNotNull(package2_keys);
|
|
|
|
Path package2Path = Paths.get(fileLocation);
|
|
byte[] header;
|
|
|
|
try (BufferedInputStream stream = new BufferedInputStream(Files.newInputStream(package2Path))){
|
|
Assertions.assertEquals(0x100, stream.skip(0x100));
|
|
header = new byte[0x100];
|
|
Assertions.assertEquals(0x100, stream.read(header));
|
|
}
|
|
|
|
Assertions.assertNotNull(header);
|
|
|
|
byte[] headerCTR = Arrays.copyOfRange(header, 0, 0x10);
|
|
|
|
for (Map.Entry<String, String> entry: package2_keys.entrySet()){
|
|
AesCtrDecryptClassic aesCtrClassic = new AesCtrDecryptClassic(entry.getValue(), headerCTR);
|
|
|
|
byte[] decrypted = aesCtrClassic.decryptNext(header);
|
|
//RainbowDump.hexDumpUTF8(decrypted);
|
|
byte[] magic = Arrays.copyOfRange(decrypted, 0x50, 0x54);
|
|
String magicString = new String(magic, StandardCharsets.US_ASCII);
|
|
if (magicString.equals("PK21"))
|
|
System.out.println(entry.getKey()+" "+entry.getValue()+" "+magicString);
|
|
}
|
|
}
|
|
@Disabled
|
|
@DisplayName("Package2 written test")
|
|
@Test
|
|
void implement() throws Exception{
|
|
System.out.printf("SIZE: %d 0x%x\n", Files.size(Paths.get(fileLocation)), Files.size(Paths.get(fileLocation)));
|
|
keyChainHolder = new KeyChainHolder(keysFileLocation, null);
|
|
System2Provider provider = new System2Provider(fileLocation, keyChainHolder);
|
|
provider.getHeader().printDebug();
|
|
provider.getKernelMap().printDebug();
|
|
Ini1Provider ini1Provider = provider.getIni1Provider();
|
|
ini1Provider.getIni1Header().printDebug();
|
|
for (Kip1 kip1 : ini1Provider.getKip1List())
|
|
kip1.printDebug();
|
|
|
|
boolean exported = provider.exportKernel("/home/loper/Projects/libKonogonka/FilesForTests/own/");
|
|
System.out.println("Exported = "+exported);
|
|
|
|
exported = ini1Provider.export("/home/loper/Projects/libKonogonka/FilesForTests/own/");
|
|
System.out.println("Exported INI1 = "+exported);
|
|
|
|
}
|
|
}
|