Compare commits

..

No commits in common. "2a3bdd949fc885fe7918c485d7d84f4d0ba5157d" and "79c519b1f3deb628df4da7a2c80b12973926b286" have entirely different histories.

33 changed files with 753 additions and 1363 deletions

24
pom.xml
View file

@ -60,28 +60,28 @@
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-controls</artifactId>
<version>16</version>
<version>11</version>
<classifier>linux</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-media</artifactId>
<version>16</version>
<version>11</version>
<classifier>linux</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-fxml</artifactId>
<version>16</version>
<version>11</version>
<classifier>linux</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-graphics</artifactId>
<version>16</version>
<version>11</version>
<classifier>linux</classifier>
<scope>compile</scope>
</dependency>
@ -89,28 +89,28 @@
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-controls</artifactId>
<version>16</version>
<version>11</version>
<classifier>win</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-media</artifactId>
<version>16</version>
<version>11</version>
<classifier>win</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-fxml</artifactId>
<version>16</version>
<version>11</version>
<classifier>win</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-graphics</artifactId>
<version>16</version>
<version>11</version>
<classifier>win</classifier>
<scope>compile</scope>
</dependency>
@ -118,28 +118,28 @@
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-controls</artifactId>
<version>16</version>
<version>11</version>
<classifier>mac</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-media</artifactId>
<version>16</version>
<version>11</version>
<classifier>mac</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-fxml</artifactId>
<version>16</version>
<version>11</version>
<classifier>mac</classifier>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.openjfx</groupId>
<artifactId>javafx-graphics</artifactId>
<version>16</version>
<version>11</version>
<classifier>mac</classifier>
<scope>compile</scope>
</dependency>

View file

@ -1,102 +0,0 @@
/*
Copyright 2019-2021 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Controllers;
import javafx.collections.ObservableList;
import javafx.fxml.FXML;
import javafx.fxml.Initializable;
import javafx.scene.control.ContextMenu;
import javafx.scene.control.ListCell;
import javafx.scene.control.ListView;
import javafx.scene.control.MenuItem;
import java.io.File;
import java.net.URL;
import java.util.List;
import java.util.ResourceBundle;
public class BlockListViewController implements Initializable {
@FXML
private ListView<File> splitMergeListView;
private ObservableList<File> filesList;
private ResourceBundle resourceBundle;
private static class FileListCell extends ListCell<File>{
@Override
public void updateItem(File file, boolean isEmpty){
super.updateItem(file, isEmpty);
if (file == null || isEmpty){
setText(null);
return;
}
String fileName = file.getName();
setText(fileName);
}
}
@Override
public void initialize(URL url, ResourceBundle resourceBundle) {
this.resourceBundle = resourceBundle;
setFilesListView();
filesList = splitMergeListView.getItems();
}
private void setFilesListView(){
splitMergeListView.setCellFactory(fileListView -> {
ListCell<File> item = new FileListCell();
setContextMenuToItem(item);
return item;
});
}
private <T> void setContextMenuToItem(ListCell<T> item){
ContextMenu contextMenu = new ContextMenu();
MenuItem deleteMenuItem = new MenuItem(resourceBundle.getString("tab1_table_contextMenu_Btn_BtnDelete"));
deleteMenuItem.setOnAction(actionEvent -> {
filesList.remove(item.getItem());
splitMergeListView.refresh();
});
MenuItem deleteAllMenuItem = new MenuItem(resourceBundle.getString("tab1_table_contextMenu_Btn_DeleteAll"));
deleteAllMenuItem.setOnAction(actionEvent -> {
filesList.clear();
splitMergeListView.refresh();
});
contextMenu.getItems().addAll(deleteMenuItem, deleteAllMenuItem);
item.setContextMenu(contextMenu);
}
public void add(File file){
if (filesList.contains(file))
return;
filesList.add(file);
}
public void addAll(List<File> files){
for (File file : files) {
add(file);
}
}
public ObservableList<File> getItems(){ return filesList; }
public void clear(){
filesList.clear();
splitMergeListView.refresh();
}
}

View file

@ -46,7 +46,7 @@ public class NSLMainController implements Initializable {
private Tab GamesTabHolder, RCMTabHolder, SMTabHolder;
@FXML
private GamesController GamesTabController;
private GamesController GamesTabController; // Accessible from Mediator | todo: incapsulate
@FXML
private SettingsController SettingsTabController;
@FXML
@ -69,12 +69,6 @@ public class NSLMainController implements Initializable {
MediatorControl.getInstance().setController(this);
if (AppPreferences.getInstance().getAutoCheckUpdates()){
checkForUpdates();
}
openLastOpenedTab();
}
private void checkForUpdates(){
Task<List<String>> updTask = new UpdatesChecker();
updTask.setOnSucceeded(event->{
List<String> result = updTask.getValue();
@ -95,6 +89,10 @@ public class NSLMainController implements Initializable {
updates.setDaemon(true);
updates.start();
}
openLastOpenedTab();
}
/**
* Get resources
* TODO: Find better solution; used in UsbCommunications() -> GL -> SelectFile command

View file

@ -18,7 +18,6 @@
*/
package nsusbloader.Controllers;
import javafx.beans.binding.Bindings;
import javafx.fxml.FXML;
import javafx.fxml.Initializable;
import javafx.scene.control.*;
@ -34,11 +33,11 @@ import nsusbloader.MediatorControl;
import nsusbloader.ModelControllers.CancellableRunnable;
import nsusbloader.NSLDataTypes.EModule;
import nsusbloader.ServiceWindow;
import nsusbloader.Utilities.splitmerge.SplitMergeTaskExecutor;
import nsusbloader.Utilities.splitmerge.MergeTask;
import nsusbloader.Utilities.splitmerge.SplitTask;
import java.io.File;
import java.net.URL;
import java.util.List;
import java.util.ResourceBundle;
public class SplitMergeController implements Initializable {
@ -54,39 +53,40 @@ public class SplitMergeController implements Initializable {
changeSaveToBtn,
convertBtn;
@FXML
private Label saveToPathLbl,
private Label fileFolderLabelLbl,
fileFolderActualPathLbl,
saveToPathLbl,
statusLbl;
@FXML
private BlockListViewController BlockListViewController;
private ResourceBundle resourceBundle;
private Region convertRegion;
private Thread smThread;
private Runnable smTask;
private CancellableRunnable smTask;
@Override
public void initialize(URL url, ResourceBundle resourceBundle) {
this.resourceBundle = resourceBundle;
convertRegion = new Region();
convertBtn.setGraphic(convertRegion);
convertBtn.disableProperty().bind(Bindings.isEmpty(BlockListViewController.getItems()));
splitRad.setOnAction((actionEvent -> {
statusLbl.setText("");
convertRegion.getStyleClass().clear();
convertRegion.getStyleClass().add("regionSplitToOne");
fileFolderLabelLbl.setText(resourceBundle.getString("tabSplMrg_Txt_File"));
selectFileFolderBtn.setText(resourceBundle.getString("tabSplMrg_Btn_SelectFile"));
BlockListViewController.clear();
fileFolderActualPathLbl.setText("");
convertBtn.setDisable(true);
}));
mergeRad.setOnAction((actionEvent -> {
statusLbl.setText("");
convertRegion.getStyleClass().clear();
convertRegion.getStyleClass().add("regionOneToSplit");
fileFolderLabelLbl.setText(resourceBundle.getString("tabSplMrg_Txt_Folder"));
selectFileFolderBtn.setText(resourceBundle.getString("tabSplMrg_Btn_SelectFolder"));
BlockListViewController.clear();
fileFolderActualPathLbl.setText("");
convertBtn.setDisable(true);
}));
if (AppPreferences.getInstance().getSplitMergeType() == 0)
@ -110,27 +110,32 @@ public class SplitMergeController implements Initializable {
selectFileFolderBtn.setOnAction(actionEvent -> {
statusLbl.setText("");
List<File> alreadyAddedFiles = BlockListViewController.getItems();
if (splitRad.isSelected()) {
FileChooser fc = new FileChooser();
fc.setTitle(resourceBundle.getString("tabSplMrg_Btn_SelectFile"));
if (! alreadyAddedFiles.isEmpty()){
String recentLocation = FilesHelper.getRealFolder(alreadyAddedFiles.get(0).getParentFile().getAbsolutePath());
fc.setInitialDirectory(new File(recentLocation));
if (! fileFolderActualPathLbl.getText().isEmpty()){
File temporaryFile = new File(fileFolderActualPathLbl.getText()).getParentFile();
if (temporaryFile != null && temporaryFile.exists())
fc.setInitialDirectory(temporaryFile);
else
fc.setInitialDirectory(new File(System.getProperty("user.home")));
}
else
fc.setInitialDirectory(new File(System.getProperty("user.home")));
List<File> files = fc.showOpenMultipleDialog(changeSaveToBtn.getScene().getWindow());
if (files == null || files.isEmpty())
File fileFile = fc.showOpenDialog(changeSaveToBtn.getScene().getWindow());
if (fileFile == null)
return;
this.BlockListViewController.addAll(files);
fileFolderActualPathLbl.setText(fileFile.getAbsolutePath());
}
else{
DirectoryChooser dc = new DirectoryChooser();
dc.setTitle(resourceBundle.getString("tabSplMrg_Btn_SelectFolder"));
if (! alreadyAddedFiles.isEmpty()){
String recentLocation = FilesHelper.getRealFolder(alreadyAddedFiles.get(0).getParentFile().getAbsolutePath());
dc.setInitialDirectory(new File(recentLocation));
if (! fileFolderActualPathLbl.getText().isEmpty()){
File temporaryFile = new File(fileFolderActualPathLbl.getText());
if (temporaryFile.exists())
dc.setInitialDirectory(temporaryFile);
else
dc.setInitialDirectory(new File(System.getProperty("user.home")));
}
else
dc.setInitialDirectory(new File(System.getProperty("user.home")));
@ -138,8 +143,9 @@ public class SplitMergeController implements Initializable {
File folderFile = dc.showDialog(changeSaveToBtn.getScene().getWindow());
if (folderFile == null)
return;
this.BlockListViewController.add(folderFile);
fileFolderActualPathLbl.setText(folderFile.getAbsolutePath());
}
convertBtn.setDisable(false);
});
convertBtn.setOnAction(actionEvent -> setConvertBtnAction());
@ -186,7 +192,7 @@ public class SplitMergeController implements Initializable {
* */
private void stopBtnAction(){
if (smThread != null && smThread.isAlive()) {
smThread.interrupt();
smTask.cancel();
}
}
/**
@ -203,9 +209,9 @@ public class SplitMergeController implements Initializable {
}
if (splitRad.isSelected())
smTask = new SplitMergeTaskExecutor(true, BlockListViewController.getItems(), saveToPathLbl.getText());
smTask = new SplitTask(fileFolderActualPathLbl.getText(), saveToPathLbl.getText());
else
smTask = new SplitMergeTaskExecutor(false, BlockListViewController.getItems(), saveToPathLbl.getText());
smTask = new MergeTask(fileFolderActualPathLbl.getText(), saveToPathLbl.getText());
smThread = new Thread(smTask);
smThread.setDaemon(true);
smThread.start();
@ -224,16 +230,14 @@ public class SplitMergeController implements Initializable {
* */
@FXML
private void handleDrop(DragEvent event) {
List<File> files = event.getDragboard().getFiles();
File firstFile = files.get(0);
File fileDrpd = event.getDragboard().getFiles().get(0);
if (firstFile.isDirectory())
if (fileDrpd.isDirectory())
mergeRad.fire();
else
splitRad.fire();
this.BlockListViewController.addAll(files);
fileFolderActualPathLbl.setText(fileDrpd.getAbsolutePath());
convertBtn.setDisable(false);
event.setDropCompleted(true);
event.consume();
}

View file

@ -26,8 +26,8 @@ import java.io.File;
import java.util.HashMap;
public interface ILogPrinter {
void print(String message, EMsgType type) throws InterruptedException;
void updateProgress(Double value) throws InterruptedException;
void print(String message, EMsgType type);
void updateProgress(Double value);
void update(HashMap<String, File> nspMap, EFileStatus status);
void update(File file, EFileStatus status);
void updateOneLinerStatus(boolean status);

View file

@ -32,11 +32,9 @@ public class LogPrinterGui implements ILogPrinter {
private final MessagesConsumer msgConsumer;
private final BlockingQueue<String> msgQueue;
private final BlockingQueue<Double> progressQueue;
private final HashMap<String, EFileStatus> statusMap;
private final HashMap<String, EFileStatus> statusMap; // BlockingQueue for literally one object. TODO: read more books ; replace to hashMap
private final AtomicBoolean oneLinerStatus;
/* TODO: Rewrite 'print()' implementation everywhere */
LogPrinterGui(EModule whoIsAsking){
this.msgQueue = new LinkedBlockingQueue<>();
this.progressQueue = new LinkedBlockingQueue<>();
@ -49,7 +47,8 @@ public class LogPrinterGui implements ILogPrinter {
* This is what will print to textArea of the application.
* */
@Override
public void print(String message, EMsgType type) throws InterruptedException{
public void print(String message, EMsgType type){
try {
switch (type){
case PASS:
msgQueue.put("[ PASS ] "+message+"\n");
@ -67,13 +66,20 @@ public class LogPrinterGui implements ILogPrinter {
msgQueue.put(message);
}
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
/**
* Update progress for progress bar
* */
@Override
public void updateProgress(Double value) throws InterruptedException {
public void updateProgress(Double value) {
try {
progressQueue.put(value);
}
catch (InterruptedException ignored){} // TODO: Do something with this
}
/**
* When we're done - update status
* */

View file

@ -42,7 +42,7 @@ public class MessagesConsumer extends AnimationTimer {
private final NSTableViewController tableViewController;
private final EModule appModuleType;
private final AtomicBoolean oneLinerStatus;
private AtomicBoolean oneLinerStatus;
private boolean isInterrupted;
@ -50,7 +50,7 @@ public class MessagesConsumer extends AnimationTimer {
BlockingQueue<String> msgQueue,
BlockingQueue<Double> progressQueue,
HashMap<String, EFileStatus> statusMap,
AtomicBoolean oneLinerStatus){
AtomicBoolean oneLinerStatus) {
this.appModuleType = appModuleType;
this.isInterrupted = false;
@ -72,15 +72,15 @@ public class MessagesConsumer extends AnimationTimer {
}
@Override
public void handle(long l){
public void handle(long l) {
ArrayList<String> messages = new ArrayList<>();
int msgReceived = msgQueue.drainTo(messages);
if (msgReceived > 0)
int msgRecieved = msgQueue.drainTo(messages);
if (msgRecieved > 0)
messages.forEach(logsArea::appendText);
ArrayList<Double> progress = new ArrayList<>();
int progressReceived = progressQueue.drainTo(progress);
if (progressReceived > 0) {
int progressRecieved = progressQueue.drainTo(progress);
if (progressRecieved > 0) {
progress.forEach(prg -> {
if (prg != 1.0)
progressBar.setProgress(prg);
@ -89,11 +89,7 @@ public class MessagesConsumer extends AnimationTimer {
});
}
if (isInterrupted) // It's safe 'cuz it's could't be interrupted while HashMap populating
updateElementsAndStop();
}
private void updateElementsAndStop(){
if (isInterrupted) { // It's safe 'cuz it's could't be interrupted while HashMap populating
MediatorControl.getInstance().setBgThreadActive(false, appModuleType);
progressBar.setProgress(0.0);
@ -113,8 +109,10 @@ public class MessagesConsumer extends AnimationTimer {
MediatorControl.getInstance().getSplitMergeController().setOneLineStatus(oneLinerStatus.get());
break;
}
this.stop();
}
}
public void interrupt(){
this.isInterrupted = true;

View file

@ -72,8 +72,8 @@ public class Rcm implements Runnable{
@Override
public void run() {
print("Selected: "+filePath, EMsgType.INFO);
print("=============== RCM ===============", EMsgType.INFO);
logPrinter.print("Selected: "+filePath, EMsgType.INFO);
logPrinter.print("=============== RCM ===============", EMsgType.INFO);
ECurrentOS ecurrentOS;
String realOsName = System.getProperty("os.name").toLowerCase().replace(" ", "");
@ -85,11 +85,11 @@ public class Rcm implements Runnable{
ecurrentOS = ECurrentOS.lin;
else
ecurrentOS = ECurrentOS.unsupported;
print("Found your OS: "+System.getProperty("os.name"), EMsgType.PASS);
logPrinter.print("Found your OS: "+System.getProperty("os.name"), EMsgType.PASS);
if (! ecurrentOS.equals(ECurrentOS.mac)){
if (! RcmSmash.isSupported()){
print("Unfortunately your platform '"+System.getProperty("os.name")+
logPrinter.print("Unfortunately your platform '"+System.getProperty("os.name")+
"' of '"+System.getProperty("os.arch")+"' is not supported :("+
"\n But you could file a bug with request."+
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL);
@ -124,14 +124,14 @@ public class Rcm implements Runnable{
// Send payload
for (int i=0; i < fullPayload.length / 4096 ; i++){
if (writeUsb(Arrays.copyOfRange(fullPayload, i*4096, (i+1)*4096))){
print("Failed to sent payload ["+i+"]"+
logPrinter.print("Failed to sent payload ["+i+"]"+
"\n\n Execution stopped.", EMsgType.FAIL);
usbConnect.close();
logPrinter.close();
return;
}
}
print("Information sent to NS.", EMsgType.PASS);
logPrinter.print("Information sent to NS.", EMsgType.PASS);
if (ecurrentOS.equals(ECurrentOS.mac)){
if (smashMacOS()){
@ -149,7 +149,7 @@ public class Rcm implements Runnable{
retval = RcmSmash.smashWindows();
else {
// ( ?_?)
print("Failed to smash the stack since your OS is not supported. Please report this issue."+
logPrinter.print("Failed to smash the stack since your OS is not supported. Please report this issue."+
"\n\n Execution stopped and failed. And it's strange.", EMsgType.FAIL);
usbConnect.close();
logPrinter.close();
@ -157,27 +157,18 @@ public class Rcm implements Runnable{
}
if (retval != 0){
print("Failed to smash the stack ("+retval+")"+
logPrinter.print("Failed to smash the stack ("+retval+")"+
"\n\n Execution stopped and failed.", EMsgType.FAIL);
usbConnect.close();
logPrinter.close();
return;
}
}
print(".:: Payload complete ::.", EMsgType.PASS);
logPrinter.print(".:: Payload complete ::.", EMsgType.PASS);
usbConnect.close();
logPrinter.updateOneLinerStatus(true);
logPrinter.close();
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException intr){
intr.printStackTrace();
}
}
/**
* Prepare the 'big' or full-size byte-buffer that is actually is a payload that we're about to use.
* @return false for issues
@ -188,7 +179,7 @@ public class Rcm implements Runnable{
// 126296 b <- biggest size per CTCaer; 16384 selected randomly as minimum threshold. It's probably wrong.
if (pldrFile.length() > 126296 || pldrFile.length() < 16384) {
print("File size of this payload looks wired. It's "+pldrFile.length()+" bytes."+
logPrinter.print("File size of this payload looks wired. It's "+pldrFile.length()+" bytes."+
"\n 1. Double-check that you're using the right payload." +
"\n 2. Please report this issue in case you're sure that you're doing everything right." +
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL);
@ -203,7 +194,7 @@ public class Rcm implements Runnable{
totalSize += 4096;
// Double-check
if (totalSize > 0x30298){
print("File size of the payload is too big. Comparing to maximum size, it's greater to "+(totalSize - 0x30298)+" bytes!"+
logPrinter.print("File size of the payload is too big. Comparing to maximum size, it's greater to "+(totalSize - 0x30298)+" bytes!"+
"\n 1. Double-check that you're using the right payload." +
"\n 2. Please report this issue in case you're sure that you're doing everything right." +
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL); // Occurs: never. I'm too lazy to check.
@ -218,7 +209,7 @@ public class Rcm implements Runnable{
BufferedInputStream bis = new BufferedInputStream(new FileInputStream(pldrFile));
int readSize;
if ((readSize = bis.read(dataPldFile)) != pldFileSize){
print("Failed to retrieve data from payload file." +
logPrinter.print("Failed to retrieve data from payload file." +
"\n Got only "+readSize+" bytes while "+pldFileSize+" expected." +
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL);
bis.close();
@ -227,7 +218,7 @@ public class Rcm implements Runnable{
bis.close();
}
catch (Exception e){
print("Failed to retrieve data from payload file: " +e.getMessage()+
logPrinter.print("Failed to retrieve data from payload file: " +e.getMessage()+
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL);
return true;
}
@ -250,7 +241,7 @@ public class Rcm implements Runnable{
IntBuffer readBufTransferred = IntBuffer.allocate(1);
int result = LibUsb.bulkTransfer(handler, (byte) 0x81, readBuffer, readBufTransferred, 1000);
if (result != LibUsb.SUCCESS) {
print("Unable to get device ID" +
logPrinter.print("Unable to get device ID" +
"\n\n Nothing has been sent to NS. Execution stopped.", EMsgType.FAIL);
return true;
}
@ -260,7 +251,7 @@ public class Rcm implements Runnable{
StringBuilder idStrBld = new StringBuilder("Found device with ID: ");
for (byte b: receivedBytes)
idStrBld.append(String.format("%02x ", b));
print(idStrBld.toString(), EMsgType.PASS);
logPrinter.print(idStrBld.toString(), EMsgType.PASS);
return false;
}
/**
@ -278,13 +269,13 @@ public class Rcm implements Runnable{
if (writeBufTransferred.get() == 4096)
return false;
print("RCM Data transfer issue [write]" +
logPrinter.print("RCM Data transfer issue [write]" +
"\n Requested: " + message.length +
"\n Transferred: " + writeBufTransferred.get()+
"\n\n Execution stopped.", EMsgType.FAIL);
return true;
}
print("RCM Data transfer issue [write]" +
logPrinter.print("RCM Data transfer issue [write]" +
"\n Returned: " + UsbErrorCodes.getErrCode(result) +
"\n\n Execution stopped.", EMsgType.FAIL);
return true;

View file

@ -38,8 +38,8 @@ public class NxdtTask extends CancellableRunnable {
@Override
public void run() {
print("Save to location: "+ saveToLocation, EMsgType.INFO);
print("=============== nxdumptool ===============", EMsgType.INFO);
logPrinter.print("Save to location: "+ saveToLocation, EMsgType.INFO);
logPrinter.print("=============== nxdumptool ===============", EMsgType.INFO);
UsbConnect usbConnect = UsbConnect.connectHomebrewMode(logPrinter);
@ -54,22 +54,13 @@ public class NxdtTask extends CancellableRunnable {
new NxdtUsbAbi1(handler, logPrinter, saveToLocation, this);
}
catch (Exception e){
print(e.getMessage(), EMsgType.FAIL);
logPrinter.print(e.getMessage(), EMsgType.FAIL);
}
print(".:: Complete ::.", EMsgType.PASS);
logPrinter.print(".:: Complete ::.", EMsgType.PASS);
usbConnect.close();
logPrinter.updateOneLinerStatus(true);
logPrinter.close();
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -119,7 +119,7 @@ class NxdtUsbAbi1 {
USBSTATUS_SUCCESS[9] = (byte)((endpointMaxPacketSize >> 8) & 0xFF);
}
private void readLoop() throws InterruptedException{
private void readLoop(){
logPrinter.print("Awaiting for handshake", EMsgType.INFO);
try {
byte[] directive;
@ -292,7 +292,7 @@ class NxdtUsbAbi1 {
return nspFile != null;
}
private String getAbsoluteFilePath(String filename) {
private String getAbsoluteFilePath(String filename) throws Exception{
if (isRomFs(filename) && isWindows) // Since RomFS entry starts from '/' it should be replaced to '\'.
return saveToPath + filename.replaceAll("/", "\\\\");
return saveToPath + filename;

View file

@ -1,178 +0,0 @@
/*
Copyright 2019-2021 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.NSLDataTypes.EMsgType;
import java.io.*;
import java.util.Arrays;
import java.util.concurrent.Callable;
public class MergeSubTask implements Callable<Boolean> {
private final int id;
private final String saveToPath;
private final MultithreadingPrintAdapter printAdapter;
private final File splitFile;
private File[] chunkFiles;
private long chunksTotalSize;
private File resultFile;
public MergeSubTask(int id, File splitFile, String saveToPath, MultithreadingPrintAdapter printAdapter){
this.id = id;
this.splitFile = splitFile;
this.saveToPath = saveToPath;
this.printAdapter = printAdapter;
}
@Override
public Boolean call(){
try{
collectChunks();
validateChunks();
sortChunks();
calculateChunksSizeSum();
createFile();
mergeChunksToFile();
validateFile();
return true;
}
catch (InterruptedException ie){
cleanup();
return false;
}
catch (Exception e){
e.printStackTrace();
try {
printAdapter.print("["+id+"] "+e.getMessage(), EMsgType.FAIL);
}
catch (InterruptedException ignore) {}
return false;
}
}
private void collectChunks(){
chunkFiles = splitFile.listFiles((file, s) -> s.matches("^[0-9][0-9]$"));
}
private void validateChunks() throws Exception{
if (chunkFiles == null || chunkFiles.length == 0){
throw new Exception("Selected folder doesn't have any chunks. Nothing to do here.");
}
}
private void sortChunks(){
Arrays.sort(chunkFiles);
}
private void calculateChunksSizeSum() throws InterruptedException{
StringBuilder builder = new StringBuilder("["+id+"] Next files will be merged in following order: ");
for (File cnk : chunkFiles){
builder.append(cnk.getName());
builder.append(" ");
chunksTotalSize += cnk.length();
}
printAdapter.print(builder.toString(), EMsgType.INFO);
}
private void createFile() throws Exception{
final String splitFileName = splitFile.getName();
resultFile = new File(saveToPath+File.separator+"!_"+splitFileName);
for (int i = 0; i < 50 ; i++){
if (interrupted())
throw new InterruptedException();
if (resultFile.exists()){
printAdapter.print("["+id+"] Trying to create a good new file...", EMsgType.WARNING);
resultFile = new File(saveToPath+File.separator+"!_"+i+"_"+splitFileName);
continue;
}
printAdapter.print("["+id+"] Save results to: "+resultFile.getAbsolutePath(), EMsgType.INFO);
return;
}
throw new Exception("Can't create new file.");
}
private void mergeChunksToFile() throws Exception{
if ( interrupted())
throw new InterruptedException();
try(BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(resultFile))){
BufferedInputStream bis;
byte[] chunk;
int readBytesCnt;
printAdapter.reportFileSize(chunksTotalSize);
for (File chunkFile : chunkFiles){
bis = new BufferedInputStream(new FileInputStream(chunkFile));
while (true){
chunk = new byte[4194240];
readBytesCnt = bis.read(chunk);
if (readBytesCnt < 4194240){
if (readBytesCnt > 0)
bos.write(chunk, 0, readBytesCnt);
break;
}
if (interrupted())
throw new InterruptedException();
bos.write(chunk);
printAdapter.updateProgressBySize(readBytesCnt);
}
bis.close();
}
}
}
private void validateFile() throws Exception{
if ( interrupted())
throw new Exception("Merge task interrupted!");
long resultFileSize = resultFile.length();
printAdapter.print("["+id+"] Total chunks size: " + chunksTotalSize
+"\n Merged file size: " + resultFileSize, EMsgType.INFO);
if (chunksTotalSize != resultFileSize)
throw new Exception("Sizes are different! Do NOT use this file for installations!");
printAdapter.print("["+id+"] Sizes are the same! Resulting file should be good!", EMsgType.PASS);
}
private void cleanup(){
boolean isDeleted = resultFile.delete();
try {
printAdapter.print(
"[" + id + "] Merge task interrupted and file "
+ (isDeleted ? "deleted." : "is NOT deleted."), EMsgType.FAIL);
}
catch (InterruptedException ignore) {}
}
private boolean interrupted(){
return Thread.interrupted(); // NOTE: it's not isInterrupted(); And it's handled properly for now.
}
}

View file

@ -0,0 +1,169 @@
/*
Copyright 2019-2020 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.ModelControllers.CancellableRunnable;
import nsusbloader.ModelControllers.ILogPrinter;
import nsusbloader.ModelControllers.Log;
import nsusbloader.NSLDataTypes.EModule;
import nsusbloader.NSLDataTypes.EMsgType;
import java.io.*;
import java.util.Arrays;
public class MergeTask extends CancellableRunnable {
private final ILogPrinter logPrinter;
private final String saveToPath;
private final String filePath;
private File splitFile;
private File[] chunkFiles;
private long chunksTotalSize;
private File resultFile;
public MergeTask(String filePath, String saveToPath) {
this.filePath = filePath;
this.saveToPath = saveToPath;
logPrinter = Log.getPrinter(EModule.SPLIT_MERGE_TOOL);
}
@Override
public void run() {
try {
logPrinter.print("Merge file: " + filePath, EMsgType.INFO);
splitFile = new File(filePath);
collectChunks();
validateChunks();
sortChunks();
calculateChunksSizeSum();
createFile();
mergeChunksToFile();
validateFile();
logPrinter.print(".:: Merge complete ::.", EMsgType.INFO);
logPrinter.updateOneLinerStatus(true);
logPrinter.close();
}
catch (Exception e){
logPrinter.print(e.getMessage(), EMsgType.FAIL);
logPrinter.updateOneLinerStatus(false);
logPrinter.close();
}
}
private void collectChunks(){
chunkFiles = splitFile.listFiles((file, s) -> s.matches("^[0-9][0-9]$"));
}
private void validateChunks() throws Exception{
if (chunkFiles == null || chunkFiles.length == 0){
throw new Exception("Selected folder doesn't have any chunks. Nothing to do here.");
}
}
private void sortChunks(){
Arrays.sort(chunkFiles);
}
private void calculateChunksSizeSum(){
logPrinter.print("Next files will be merged in following order: ", EMsgType.INFO);
for (File cnk : chunkFiles){
logPrinter.print(" "+cnk.getName(), EMsgType.INFO);
chunksTotalSize += cnk.length();
}
}
private void createFile() throws Exception{
final String splitFileName = splitFile.getName();
resultFile = new File(saveToPath+File.separator+"!_"+splitFileName);
for (int i = 0; i < 50 ; i++){
if (isCancelled()){
throw new InterruptedException("Split task interrupted!");
}
if (resultFile.exists()){
logPrinter.print("Trying to create a good new file...", EMsgType.WARNING);
resultFile = new File(saveToPath+File.separator+"!_"+i+"_"+splitFileName);
continue;
}
logPrinter.print("Save results to: "+resultFile.getAbsolutePath(), EMsgType.INFO);
return;
}
throw new Exception("Can't create new file.");
}
private void mergeChunksToFile() throws Exception{
double chunkPercent = (4194240.0 / (chunksTotalSize / 100.0) / 100.0);
long totalSizeCnt = 0;
BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(resultFile));
BufferedInputStream bis;
byte[] chunk;
int readBytesCnt;
for (File chunkFile : chunkFiles){
bis = new BufferedInputStream(new FileInputStream(chunkFile));
while (true){
if (isCancelled()){
bos.close();
bis.close();
boolean isDeleted = resultFile.delete();
throw new InterruptedException("Merge task interrupted and file "
+ (isDeleted ? "deleted." : "is not deleted."));
}
chunk = new byte[4194240];
readBytesCnt = bis.read(chunk);
logPrinter.updateProgress(chunkPercent * totalSizeCnt);
totalSizeCnt++;
if (readBytesCnt < 4194240){
if (readBytesCnt > 0)
bos.write(chunk, 0, readBytesCnt);
break;
}
bos.write(chunk);
}
bis.close();
}
bos.close();
}
private void validateFile() throws Exception{
long resultFileSize = resultFile.length();
logPrinter.print("Total chunks size: " + chunksTotalSize, EMsgType.INFO);
logPrinter.print("Merged file size: " + resultFileSize, EMsgType.INFO);
if (chunksTotalSize != resultFileSize)
throw new Exception("Sizes are different! Do NOT use this file for installations!");
logPrinter.print("Sizes are the same! Resulting file should be good!", EMsgType.PASS);
}
}

View file

@ -1,44 +0,0 @@
/*
Copyright 2019-2021 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.ModelControllers.ILogPrinter;
import nsusbloader.NSLDataTypes.EMsgType;
public class MultithreadingPrintAdapter {
private final ILogPrinter printer;
private long totalFilesSize;
private long bytesComplete;
public MultithreadingPrintAdapter(ILogPrinter printer){
this.printer = printer;
}
public void print(String message, EMsgType type) throws InterruptedException{
printer.print(message, type);
}
public void reportFileSize(long fileSize){
totalFilesSize += fileSize;
}
public void updateProgressBySize(long chunkSize) throws InterruptedException{
bytesComplete += chunkSize;
printer.updateProgress((double) bytesComplete / (double) totalFilesSize);
}
}

View file

@ -1,147 +0,0 @@
/*
Copyright 2019-2020 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.ModelControllers.ILogPrinter;
import nsusbloader.ModelControllers.Log;
import nsusbloader.NSLDataTypes.EModule;
import nsusbloader.NSLDataTypes.EMsgType;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.*;
/*
* TODO: Kill this on application exit (?)
*/
public class SplitMergeTaskExecutor implements Runnable {
private final boolean isSplit;
private final List<File> files;
private final String saveToPath;
private final ILogPrinter logPrinter;
private final ExecutorService executorService;
private final MultithreadingPrintAdapter printAdapter;
public SplitMergeTaskExecutor(boolean isSplit, List<File> files, String saveToPath){
this.isSplit = isSplit;
this.files = files;
this.saveToPath = saveToPath;
this.logPrinter = Log.getPrinter(EModule.SPLIT_MERGE_TOOL);
this.executorService = Executors.newFixedThreadPool(
files.size(),
runnable -> {
Thread thread = new Thread(runnable);
thread.setDaemon(true);
return thread;
});
this.printAdapter = new MultithreadingPrintAdapter(logPrinter);
}
public void run(){
try {
List<Future<Boolean>> futuresResults = executorService.invokeAll(getSubTasksCollection());
boolean onelinerResult = true;
for (Future<Boolean> future : futuresResults){
onelinerResult &= future.get();
}
executorService.shutdown();
logPrinter.updateOneLinerStatus(onelinerResult);
}
catch (InterruptedException ie){
//ie.printStackTrace();
executorService.shutdownNow();
boolean interruptedSuccessfully = false;
try {
interruptedSuccessfully = executorService.awaitTermination(20, TimeUnit.SECONDS);
}
catch (InterruptedException awaitInterrupt){
print("Force interrupting task...", EMsgType.WARNING);
}
logPrinter.updateOneLinerStatus(false);
print((
isSplit?
"Split tasks interrupted ":
"Merge tasks interrupted ")+
(interruptedSuccessfully?
"successfully":
"with some issues"), EMsgType.WARNING);
}
catch (Exception e){
logPrinter.updateOneLinerStatus(false);
print(
isSplit?
"Split task failed: ":
"Merge task failed: "+e.getMessage(), EMsgType.FAIL);
e.printStackTrace();
}
print(
isSplit?
".:: Split complete ::.":
".:: Merge complete ::.", EMsgType.INFO);
logPrinter.close();
}
private List<Callable<Boolean>> getSubTasksCollection() throws InterruptedException{
List<Callable<Boolean>> subTasks = new ArrayList<>();
StringBuilder stringBuilder = new StringBuilder();
// TODO: Optimize?
if (isSplit){
stringBuilder.append("Split files:\n");
for (int i = 0; i < files.size(); i++){
File file = files.get(i);
stringBuilder.append("[");
stringBuilder.append(i);
stringBuilder.append("] ");
stringBuilder.append(file.getName());
stringBuilder.append("\n");
Callable<Boolean> task = new SplitSubTask(i, file, saveToPath, printAdapter);
subTasks.add(task);
}
}
else {
stringBuilder.append("Merge files:\n");
for (int i = 0; i < files.size(); i++){
File file = files.get(i);
stringBuilder.append("[");
stringBuilder.append(i);
stringBuilder.append("] ");
stringBuilder.append(file.getName());
stringBuilder.append("\n");
Callable<Boolean> task = new MergeSubTask(i, file, saveToPath, printAdapter);
subTasks.add(task);
}
}
logPrinter.print(stringBuilder.toString(), EMsgType.INFO);
return subTasks;
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -1,186 +0,0 @@
/*
Copyright 2019-2021 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.NSLDataTypes.EMsgType;
import java.io.*;
import java.util.Arrays;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
public class SplitSubTask implements Callable<Boolean> {
private final int id;
private final String saveToPath;
private final File file;
private File splitFile;
private long originalFileLen;
private final MultithreadingPrintAdapter printAdapter;
SplitSubTask(int id, File file, String saveToPath, MultithreadingPrintAdapter printAdapter){
this.id = id;
this.file = file;
this.saveToPath = saveToPath;
this.printAdapter = printAdapter;
}
@Override
public Boolean call(){
try {
createSplitFile();
splitFileToChunks();
validateSplitFile();
return true;
}
catch (InterruptedException | ExecutionException ie){
ie.printStackTrace();
cleanup();
return false;
}
catch (Exception e){
e.printStackTrace();
try {
printAdapter.print("["+id+"] "+e.getMessage(), EMsgType.FAIL);
}
catch (InterruptedException ignore) {}
return false;
}
}
private void createSplitFile() throws Exception{
if ( interrupted())
throw new Exception("Split task interrupted!");
splitFile = new File(saveToPath+File.separator+"!_"+file.getName());
for (int i = 0; i < 50; i++){
if (splitFile.mkdirs()){
printAdapter.print("["+id+"] Save results to: "+splitFile.getAbsolutePath(), EMsgType.INFO);
return;
}
if (splitFile.exists()){
printAdapter.print("["+id+"] Trying to create a good new folder...", EMsgType.WARNING);
splitFile = new File(saveToPath+File.separator+"!_"+i+"_"+file.getName());
continue;
}
throw new Exception("Folder " + splitFile.getAbsolutePath()
+ " could not be created. Not enough rights or something like that?");
}
throw new Exception("Can't create new file.");
}
private void splitFileToChunks() throws Exception{
try(BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file))){
long counter;
originalFileLen = file.length();
printAdapter.reportFileSize(originalFileLen);
byte[] chunk;
int readBytesCnt;
main_loop:
for (int i = 0; ; i++){
String pathname = splitFile.getAbsolutePath()+File.separator+String.format("%02d", i);
BufferedOutputStream fragmentBos = new BufferedOutputStream(new FileOutputStream(pathname));
counter = 0;
while (counter < 1024){ // 0xffff0000 total
chunk = new byte[4194240];
if ((readBytesCnt = bis.read(chunk)) < 4194240){
if (readBytesCnt > 0)
fragmentBos.write(chunk, 0, readBytesCnt);
fragmentBos.close();
printAdapter.updateProgressBySize(readBytesCnt);
break main_loop;
}
if (interrupted())
throw new InterruptedException();
fragmentBos.write(chunk);
counter++;
printAdapter.updateProgressBySize(readBytesCnt);
}
fragmentBos.close();
}
}
}
private void validateSplitFile() throws Exception{
if (interrupted())
throw new Exception("Split task interrupted!");
printAdapter.print("["+id+"] Original file: "+splitFile.getAbsolutePath()+" (size: "+originalFileLen+")", EMsgType.INFO);
long totalChunksSize = 0;
File[] chunkFileArr = splitFile.listFiles();
if (chunkFileArr == null)
throw new Exception("Unable to check results. It means that something went wrong.");
Arrays.sort(chunkFileArr);
StringBuilder stringBuilder = new StringBuilder("["+id+"] Chunks");
for (File chunkFile : chunkFileArr) {
stringBuilder.append("\n");
stringBuilder.append(" ");
stringBuilder.append(chunkFile.getName());
stringBuilder.append(" size: ");
stringBuilder.append(chunkFile.length());
totalChunksSize += chunkFile.length();
}
stringBuilder.append("\n");
stringBuilder.append("Total chunks size: ");
stringBuilder.append(totalChunksSize);
printAdapter.print(stringBuilder.toString(), EMsgType.INFO);
if (originalFileLen != totalChunksSize)
throw new Exception("Sizes are different! Do NOT use this file for installations!");
printAdapter.print("["+id+"] Sizes are the same! Split file should be good!", EMsgType.PASS);
}
private void cleanup(){
boolean isDeleted = splitFile.delete();
File[] chunksToDelete = splitFile.listFiles();
if (! isDeleted && chunksToDelete != null){
isDeleted = true;
for (File chunkFile : chunksToDelete)
isDeleted &= chunkFile.delete();
isDeleted &= splitFile.delete();
}
try {
printAdapter.print(
"["+id+"] Split task interrupted and folder "
+ (isDeleted?"deleted.":"is NOT deleted.")
, EMsgType.FAIL);
}
catch (InterruptedException ignore) {}
}
private boolean interrupted(){
return Thread.interrupted(); // NOTE: it's not isInterrupted(); And it's handled properly for now.
}
}

View file

@ -0,0 +1,173 @@
/*
Copyright 2019-2020 Dmitry Isaenko
This file is part of NS-USBloader.
NS-USBloader 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.
NS-USBloader 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 NS-USBloader. If not, see <https://www.gnu.org/licenses/>.
*/
package nsusbloader.Utilities.splitmerge;
import nsusbloader.ModelControllers.CancellableRunnable;
import nsusbloader.ModelControllers.ILogPrinter;
import nsusbloader.ModelControllers.Log;
import nsusbloader.NSLDataTypes.EModule;
import nsusbloader.NSLDataTypes.EMsgType;
import java.io.*;
import java.util.Arrays;
public class SplitTask extends CancellableRunnable {
private final ILogPrinter logPrinter;
private final String saveToPath;
private final String filePath;
private File file;
private File splitFile;
private long originalFileLen;
public SplitTask(String filePath, String saveToPath){
this.filePath = filePath;
this.saveToPath = saveToPath;
this.logPrinter = Log.getPrinter(EModule.SPLIT_MERGE_TOOL);
}
@Override
public void run() {
try {
logPrinter.print("Split file: "+filePath, EMsgType.INFO);
this.file = new File(filePath);
createSplitFile();
splitFileToChunks();
validateSplitFile();
logPrinter.print(".:: Split complete ::.", EMsgType.INFO);
logPrinter.updateOneLinerStatus(true);
logPrinter.close();
}
catch (Exception e){
logPrinter.print(e.getMessage(), EMsgType.FAIL);
logPrinter.updateOneLinerStatus(false);
logPrinter.close();
}
}
private void createSplitFile() throws Exception{
splitFile = new File(saveToPath+File.separator+"!_"+file.getName());
for (int i = 0; i < 50 ; i++){
if (isCancelled()){
throw new InterruptedException("Split task interrupted!");
}
if (splitFile.mkdir()){
logPrinter.print("Save results to: "+splitFile.getAbsolutePath(), EMsgType.INFO);
return;
}
if (splitFile.exists()){
logPrinter.print("Trying to create a good new folder...", EMsgType.WARNING);
splitFile = new File(saveToPath+File.separator+"!_"+i+"_"+file.getName());
continue;
}
throw new Exception("Folder " + splitFile.getAbsolutePath()
+ " could not be created. Not enough rights or something like that?");
}
throw new Exception("Can't create new file.");
}
private void splitFileToChunks() throws Exception{
BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
long counter;
originalFileLen = file.length();
double chunkPercent = (4194240.0 / (originalFileLen / 100.0) / 100.0);
long totalSizeCnt = 0;
byte[] chunk;
int readBytesCnt;
main_loop:
for (int i = 0; ; i++){
String pathname = splitFile.getAbsolutePath()+File.separator+String.format("%02d", i);
BufferedOutputStream fragmentBos = new BufferedOutputStream(new FileOutputStream(new File(pathname)));
counter = 0;
while (counter < 1024){ // 0xffff0000 total
if (isCancelled()){
fragmentBos.close();
bis.close();
boolean isDeleted = splitFile.delete();
File[] chunksToDelete = splitFile.listFiles();
if (! isDeleted && chunksToDelete != null){
isDeleted = true;
for (File chunkFile : chunksToDelete)
isDeleted &= chunkFile.delete();
isDeleted &= splitFile.delete();
}
throw new InterruptedException("Split task interrupted and folder "
+ (isDeleted?"deleted.":"is not deleted."));
}
chunk = new byte[4194240];
if ((readBytesCnt = bis.read(chunk)) < 4194240){
if (readBytesCnt > 0)
fragmentBos.write(chunk, 0, readBytesCnt);
fragmentBos.close();
logPrinter.updateProgress(1.0);
break main_loop;
}
fragmentBos.write(chunk);
logPrinter.updateProgress(chunkPercent * totalSizeCnt);
counter++; // NOTE: here we have some redundancy of variables. It has to be fixed one day.
totalSizeCnt++;
}
fragmentBos.close();
}
bis.close();
}
private void validateSplitFile() throws Exception{
logPrinter.print("Original file size: "+originalFileLen, EMsgType.INFO);
long totalChunksSize = 0;
File[] chunkFileArr = splitFile.listFiles();
if (chunkFileArr == null)
throw new Exception("Unable to check results. It means that something went wrong.");
Arrays.sort(chunkFileArr);
for (File chunkFile : chunkFileArr) {
logPrinter.print("Chunk " + chunkFile.getName() + " size: " + chunkFile.length(), EMsgType.INFO);
totalChunksSize += chunkFile.length();
}
logPrinter.print("Total chunks size: " + totalChunksSize, EMsgType.INFO);
if (originalFileLen != totalChunksSize)
throw new Exception("Sizes are different! Do NOT use this file for installations!");
logPrinter.print("Sizes are the same! Split file should be good!", EMsgType.PASS);
}
}

View file

@ -18,7 +18,8 @@
*/
package nsusbloader.cli;
import nsusbloader.Utilities.splitmerge.SplitMergeTaskExecutor;
import nsusbloader.Utilities.splitmerge.MergeTask;
import nsusbloader.Utilities.splitmerge.SplitTask;
import java.io.File;
import java.util.ArrayList;
@ -26,7 +27,7 @@ import java.util.List;
public class MergeCli {
private final String[] arguments;
private String[] arguments;
private String saveTo;
private String[] splitFiles;
@ -96,20 +97,12 @@ public class MergeCli {
}
private void runBackend() throws InterruptedException{
Runnable mergeTask = new SplitMergeTaskExecutor(
false,
getFilesFromStrings(),
saveTo);
for (String filePath : splitFiles){
Runnable mergeTask = new MergeTask(filePath, saveTo);
Thread thread = new Thread(mergeTask);
thread.setDaemon(true);
thread.start();
thread.join();
}
private List<File> getFilesFromStrings(){
ArrayList<File> realFiles = new ArrayList<>();
for (String splitFileString : splitFiles){
realFiles.add(new File(splitFileString));
}
return realFiles;
}
}

View file

@ -18,14 +18,16 @@
*/
package nsusbloader.cli;
import nsusbloader.Utilities.splitmerge.SplitMergeTaskExecutor;
import nsusbloader.Utilities.splitmerge.SplitTask;
import java.io.File;
import java.util.*;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
public class SplitCli {
private final String[] arguments;
private String[] arguments;
private String saveTo;
private String[] files;
@ -93,20 +95,12 @@ public class SplitCli {
}
private void runBackend() throws InterruptedException{
Runnable splitTasks = new SplitMergeTaskExecutor(
true,
getFilesFromStrings(),
saveTo);
Thread thread = new Thread(splitTasks);
for (String filePath : files){
Runnable splitTaks = new SplitTask(filePath, saveTo);
Thread thread = new Thread(splitTaks);
thread.setDaemon(true);
thread.start();
thread.join();
}
private List<File> getFilesFromStrings(){
ArrayList<File> realFiles = new ArrayList<>();
for (String fileString : files){
realFiles.add(new File(fileString));
}
return realFiles;
}
}

View file

@ -90,7 +90,7 @@ public class NETCommunications extends CancellableRunnable {
if (! isValid || isCancelled() )
return;
print("\tStart chain", EMsgType.INFO);
logPrinter.print("\tStart chain", EMsgType.INFO);
final String handshakeContent = buildHandshakeContent();
@ -102,11 +102,11 @@ public class NETCommunications extends CancellableRunnable {
// Check if we should serve requests
if (this.doNotServe){
print("List of files transferred. Replies won't be served.", EMsgType.PASS);
logPrinter.print("List of files transferred. Replies won't be served.", EMsgType.PASS);
close(EFileStatus.UNKNOWN);
return;
}
print("Initiation files list has been sent to NS.", EMsgType.PASS);
logPrinter.print("Initiation files list has been sent to NS.", EMsgType.PASS);
// Go transfer
serveRequestsLoop();
@ -142,7 +142,7 @@ public class NETCommunications extends CancellableRunnable {
handshakeSocket.close();
}
catch (IOException uhe){
print("Unable to connect to NS and send files list:\n "
logPrinter.print("Unable to connect to NS and send files list:\n "
+ uhe.getMessage(), EMsgType.FAIL);
close(EFileStatus.UNKNOWN);
return true;
@ -175,13 +175,13 @@ public class NETCommunications extends CancellableRunnable {
}
catch (Exception e){
if (isCancelled())
print("Interrupted by user.", EMsgType.INFO);
logPrinter.print("Interrupted by user.", EMsgType.INFO);
else
print(e.getMessage(), EMsgType.INFO);
logPrinter.print(e.getMessage(), EMsgType.INFO);
close(EFileStatus.UNKNOWN);
return;
}
print("All transfers complete", EMsgType.PASS);
logPrinter.print("All transfers complete", EMsgType.PASS);
close(EFileStatus.UPLOADED);
}
/**
@ -199,7 +199,7 @@ public class NETCommunications extends CancellableRunnable {
if (! files.containsKey(reqFileName)){
writeToSocket(NETPacket.getCode404());
print("File "+reqFileName+" doesn't exists or have 0 size. Returning 404", EMsgType.FAIL);
logPrinter.print("File "+reqFileName+" doesn't exists or have 0 size. Returning 404", EMsgType.FAIL);
return;
}
@ -208,13 +208,13 @@ public class NETCommunications extends CancellableRunnable {
if (! requestedFile.exists() || reqFileSize == 0){ // well.. tell 404 if file exists with 0 length is against standard, but saves time
writeToSocket(NETPacket.getCode404());
print("File "+requestedFile.getName()+" doesn't exists or have 0 size. Returning 404", EMsgType.FAIL);
logPrinter.print("File "+requestedFile.getName()+" doesn't exists or have 0 size. Returning 404", EMsgType.FAIL);
logPrinter.update(requestedFile, EFileStatus.FAILED);
return;
}
if (packet.get(0).startsWith("HEAD")){
writeToSocket(NETPacket.getCode200(reqFileSize));
print("Replying for requested file: "+requestedFile.getName(), EMsgType.INFO);
logPrinter.print("Replying for requested file: "+requestedFile.getName(), EMsgType.INFO);
return;
}
if (packet.get(0).startsWith("GET")) {
@ -237,7 +237,7 @@ public class NETCommunications extends CancellableRunnable {
if (fromRange > toRange){ // If start bytes greater then end bytes
writeToSocket(NETPacket.getCode400());
print("Requested range for "
logPrinter.print("Requested range for "
+ file.getName()
+ " is incorrect. Returning 400", EMsgType.FAIL);
logPrinter.update(file, EFileStatus.FAILED);
@ -254,7 +254,7 @@ public class NETCommunications extends CancellableRunnable {
if (rangeStr[1].isEmpty()) { // If Range not defined: like "Range: bytes=-"
writeToSocket(NETPacket.getCode400());
print("Requested range for "
logPrinter.print("Requested range for "
+ file.getName()
+ " is incorrect (empty start & end). Returning 400", EMsgType.FAIL);
logPrinter.update(file, EFileStatus.FAILED);
@ -267,7 +267,7 @@ public class NETCommunications extends CancellableRunnable {
}
// If file smaller than 500 bytes
writeToSocket(NETPacket.getCode416());
print("File size requested for "
logPrinter.print("File size requested for "
+ file.getName()
+ " while actual size of it: "
+ fileSize+". Returning 416", EMsgType.FAIL);
@ -275,7 +275,7 @@ public class NETCommunications extends CancellableRunnable {
}
catch (NumberFormatException nfe){
writeToSocket(NETPacket.getCode400());
print("Requested range for "
logPrinter.print("Requested range for "
+ file.getName()
+ " has incorrect format. Returning 400\n\t"
+ nfe.getMessage(), EMsgType.FAIL);
@ -293,7 +293,7 @@ public class NETCommunications extends CancellableRunnable {
private void writeToSocket(String fileName, long start, long end) throws Exception{
File file = files.get(fileName).getFile();
print("Reply to range: "+start+"-"+end, EMsgType.INFO);
logPrinter.print("Reply to range: "+start+"-"+end, EMsgType.INFO);
writeToSocket(NETPacket.getCode206(files.get(fileName).getSize(), start, end));
try{
@ -379,11 +379,11 @@ public class NETCommunications extends CancellableRunnable {
try {
if (serverSocket != null && ! serverSocket.isClosed()) {
serverSocket.close();
print("Closing server socket.", EMsgType.PASS);
logPrinter.print("Closing server socket.", EMsgType.PASS);
}
}
catch (IOException ioe){
print("Closing server socket failed. Sometimes it's not an issue.", EMsgType.WARNING);
logPrinter.print("Closing server socket failed. Sometimes it's not an issue.", EMsgType.WARNING);
}
HashMap<String, File> tempMap = new HashMap<>();
@ -392,15 +392,7 @@ public class NETCommunications extends CancellableRunnable {
logPrinter.update(tempMap, status);
print("\tEnd chain", EMsgType.INFO);
logPrinter.print("\tEnd chain", EMsgType.INFO);
logPrinter.close();
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -55,21 +55,15 @@ public class NetworkSetupValidator {
resolvePort(hostPortNum);
}
catch (Exception e){
try {
logPrinter.print(e.getMessage(), EMsgType.FAIL);
}
catch (InterruptedException ignore){}
valid = false;
return;
}
valid = true;
}
private void validateFiles(List<File> filesList){
filesList.removeIf(this::validator);
}
private boolean validator(File f){
try {
private void validateFiles(List<File> filesList) {
filesList.removeIf(f -> {
if (f.isFile())
return false;
@ -82,27 +76,24 @@ public class NetworkSetupValidator {
Arrays.sort(subFiles, Comparator.comparingInt(file -> Integer.parseInt(file.getName())));
for (int i = subFiles.length - 2; i > 0; i--) {
if (subFiles[i].length() != subFiles[i - 1].length()) {
logPrinter.print("NET: Exclude split file: " + f.getName() +
for (int i = subFiles.length - 2; i > 0 ; i--){
if (subFiles[i].length() != subFiles[i-1].length()) {
logPrinter.print("NET: Exclude split file: "+f.getName()+
"\n Chunk sizes of the split file are not the same, but has to be.", EMsgType.WARNING);
return true;
}
}
long firstFileLength = subFiles[0].length();
long lastFileLength = subFiles[subFiles.length - 1].length();
long lastFileLength = subFiles[subFiles.length-1].length();
if (lastFileLength > firstFileLength) {
logPrinter.print("NET: Exclude split file: " + f.getName() +
if (lastFileLength > firstFileLength){
logPrinter.print("NET: Exclude split file: "+f.getName()+
"\n Chunk sizes of the split file are not the same, but has to be.", EMsgType.WARNING);
return true;
}
return false;
}
catch (InterruptedException ie){
return false;
}
});
}
private void encodeAndAddFilesToMap(List<File> filesList) throws UnsupportedEncodingException, FileNotFoundException {
@ -117,7 +108,7 @@ public class NetworkSetupValidator {
}
}
private void resolveIp(String hostIPaddr) throws IOException, InterruptedException{
private void resolveIp(String hostIPaddr) throws IOException{
if (! hostIPaddr.isEmpty()){
this.hostIP = hostIPaddr;
logPrinter.print("NET: Host IP defined as: " + hostIP, EMsgType.PASS);
@ -133,7 +124,7 @@ public class NetworkSetupValidator {
throw new IOException("Try using 'Expert mode' and set IP manually. " + getAvaliableIpExamples());
}
private boolean findIpUsingHost(String host) throws InterruptedException{
private boolean findIpUsingHost(String host) {
try {
Socket scoketK;
scoketK = new Socket();

View file

@ -59,17 +59,17 @@ public class GoldLeaf_05 extends TransferModule {
this.task = task;
status = EFileStatus.FAILED;
print("============= GoldLeaf v0.5 =============\n" +
logPrinter.print("============= GoldLeaf v0.5 =============\n" +
" Only one file per time could be sent. In case you selected more the first one would be picked.", EMsgType.INFO);
if (nspMap.isEmpty()){
print("For using this GoldLeaf version you have to add file to the table and select it for upload", EMsgType.INFO);
logPrinter.print("For using this GoldLeaf version you have to add file to the table and select it for upload", EMsgType.INFO);
return;
}
File nspFile = (File) nspMap.values().toArray()[0];
print("File for upload: "+nspFile.getAbsolutePath(), EMsgType.INFO);
logPrinter.print("File for upload: "+nspFile.getAbsolutePath(), EMsgType.INFO);
if (!nspFile.getName().toLowerCase().endsWith(".nsp")) {
print("GL This file doesn't look like NSP", EMsgType.FAIL);
logPrinter.print("GL This file doesn't look like NSP", EMsgType.FAIL);
return;
}
PFSProvider pfsElement;
@ -77,11 +77,11 @@ public class GoldLeaf_05 extends TransferModule {
pfsElement = new PFSProvider(nspFile, logPrinter);
}
catch (Exception e){
print("GL File provided has incorrect structure and won't be uploaded\n\t"+e.getMessage(), EMsgType.FAIL);
logPrinter.print("GL File provided has incorrect structure and won't be uploaded\n\t"+e.getMessage(), EMsgType.FAIL);
status = EFileStatus.INCORRECT_FILE_FAILED;
return;
}
print("GL File structure validated and it will be uploaded", EMsgType.PASS);
logPrinter.print("GL File structure validated and it will be uploaded", EMsgType.PASS);
try{
if (nspFile.isDirectory())
@ -90,7 +90,7 @@ public class GoldLeaf_05 extends TransferModule {
this.raf = new RandomAccessFile(nspFile, "r");
}
catch (IOException ioe){
print("GL File not found\n\t"+ioe.getMessage(), EMsgType.FAIL);
logPrinter.print("GL File not found\n\t"+ioe.getMessage(), EMsgType.FAIL);
return;
}
@ -99,15 +99,15 @@ public class GoldLeaf_05 extends TransferModule {
// Go connect to GoldLeaf
if (writeUsb(CMD_GLUC)) {
print("GL Initiating GoldLeaf connection [1/2]", EMsgType.FAIL);
logPrinter.print("GL Initiating GoldLeaf connection [1/2]", EMsgType.FAIL);
return;
}
print("GL Initiating GoldLeaf connection: [1/2]", EMsgType.PASS);
logPrinter.print("GL Initiating GoldLeaf connection: [1/2]", EMsgType.PASS);
if (writeUsb(CMD_ConnectionRequest)){
print("GL Initiating GoldLeaf connection: [2/2]", EMsgType.FAIL);
logPrinter.print("GL Initiating GoldLeaf connection: [2/2]", EMsgType.FAIL);
return;
}
print("GL Initiating GoldLeaf connection: [2/2]", EMsgType.PASS);
logPrinter.print("GL Initiating GoldLeaf connection: [2/2]", EMsgType.PASS);
while (true) {
readByte = readUsb();
@ -143,7 +143,7 @@ public class GoldLeaf_05 extends TransferModule {
continue;
}
if (Arrays.equals(readByte, CMD_Finish)) {
print("GL Closing GoldLeaf connection: Transfer successful.", EMsgType.PASS);
logPrinter.print("GL Closing GoldLeaf connection: Transfer successful.", EMsgType.PASS);
status = EFileStatus.UPLOADED;
break;
}
@ -164,29 +164,29 @@ public class GoldLeaf_05 extends TransferModule {
* false if no issues
* */
private boolean handleConnectionResponse(PFSProvider pfsElement){
print("GL 'ConnectionResponse' command:", EMsgType.INFO);
logPrinter.print("GL 'ConnectionResponse' command:", EMsgType.INFO);
if (writeUsb(CMD_GLUC)) {
print(" [1/4]", EMsgType.FAIL);
logPrinter.print(" [1/4]", EMsgType.FAIL);
return true;
}
print(" [1/4]", EMsgType.PASS);
logPrinter.print(" [1/4]", EMsgType.PASS);
if (writeUsb(CMD_NSPName)) {
print(" [2/4]", EMsgType.FAIL);
logPrinter.print(" [2/4]", EMsgType.FAIL);
return true;
}
print(" [2/4]", EMsgType.PASS);
logPrinter.print(" [2/4]", EMsgType.PASS);
if (writeUsb(pfsElement.getBytesNspFileNameLength())) {
print(" [3/4]", EMsgType.FAIL);
logPrinter.print(" [3/4]", EMsgType.FAIL);
return true;
}
print(" [3/4]", EMsgType.PASS);
logPrinter.print(" [3/4]", EMsgType.PASS);
if (writeUsb(pfsElement.getBytesNspFileName())) {
print(" [4/4]", EMsgType.FAIL);
logPrinter.print(" [4/4]", EMsgType.FAIL);
return true;
}
print(" [4/4]", EMsgType.PASS);
logPrinter.print(" [4/4]", EMsgType.PASS);
return false;
}
@ -196,50 +196,50 @@ public class GoldLeaf_05 extends TransferModule {
* false if no issues
* */
private boolean handleStart(PFSProvider pfsElement){
print("GL Handle 'Start' command:", EMsgType.INFO);
logPrinter.print("GL Handle 'Start' command:", EMsgType.INFO);
if (writeUsb(CMD_GLUC)) {
print(" [Prefix]", EMsgType.FAIL);
logPrinter.print(" [Prefix]", EMsgType.FAIL);
return true;
}
print(" [Prefix]", EMsgType.PASS);
logPrinter.print(" [Prefix]", EMsgType.PASS);
if (writeUsb(CMD_NSPData)) {
print(" [Command]", EMsgType.FAIL);
logPrinter.print(" [Command]", EMsgType.FAIL);
return true;
}
print(" [Command]", EMsgType.PASS);
logPrinter.print(" [Command]", EMsgType.PASS);
if (writeUsb(pfsElement.getBytesCountOfNca())) {
print(" [Sub-files count]", EMsgType.FAIL);
logPrinter.print(" [Sub-files count]", EMsgType.FAIL);
return true;
}
print(" [Sub-files count]", EMsgType.PASS);
logPrinter.print(" [Sub-files count]", EMsgType.PASS);
int ncaCount = pfsElement.getIntCountOfNca();
print(" [Information for "+ncaCount+" sub-files]", EMsgType.INFO);
logPrinter.print(" [Information for "+ncaCount+" sub-files]", EMsgType.INFO);
for (int i = 0; i < ncaCount; i++){
print("File #"+i, EMsgType.INFO);
logPrinter.print("File #"+i, EMsgType.INFO);
if (writeUsb(pfsElement.getNca(i).getNcaFileNameLength())) {
print(" [1/4] Name length", EMsgType.FAIL);
logPrinter.print(" [1/4] Name length", EMsgType.FAIL);
return true;
}
print(" [1/4] Name length", EMsgType.PASS);
logPrinter.print(" [1/4] Name length", EMsgType.PASS);
if (writeUsb(pfsElement.getNca(i).getNcaFileName())) {
print(" [2/4] Name", EMsgType.FAIL);
logPrinter.print(" [2/4] Name", EMsgType.FAIL);
return true;
}
print(" [2/4] Name", EMsgType.PASS);
logPrinter.print(" [2/4] Name", EMsgType.PASS);
if (writeUsb(ByteBuffer.allocate(8).order(ByteOrder.LITTLE_ENDIAN).putLong(pfsElement.getBodySize()+pfsElement.getNca(i).getNcaOffset()).array())) { // offset. real.
print(" [3/4] Offset", EMsgType.FAIL);
logPrinter.print(" [3/4] Offset", EMsgType.FAIL);
return true;
}
print(" [3/4] Offset", EMsgType.PASS);
logPrinter.print(" [3/4] Offset", EMsgType.PASS);
if (writeUsb(ByteBuffer.allocate(8).order(ByteOrder.LITTLE_ENDIAN).putLong(pfsElement.getNca(i).getNcaSize()).array())) { // size
print(" [4/4] Size", EMsgType.FAIL);
logPrinter.print(" [4/4] Size", EMsgType.FAIL);
return true;
}
print(" [4/4] Size", EMsgType.PASS);
logPrinter.print(" [4/4] Size", EMsgType.PASS);
}
return false;
}
@ -254,18 +254,18 @@ public class GoldLeaf_05 extends TransferModule {
int requestedNcaID;
if (isItRawRequest) {
print("GL Handle 'Content' command", EMsgType.INFO);
logPrinter.print("GL Handle 'Content' command", EMsgType.INFO);
byte[] readByte = readUsb();
if (readByte == null || readByte.length != 4) {
print(" [Read requested ID]", EMsgType.FAIL);
logPrinter.print(" [Read requested ID]", EMsgType.FAIL);
return true;
}
requestedNcaID = ByteBuffer.wrap(readByte).order(ByteOrder.LITTLE_ENDIAN).getInt();
print(" [Read requested ID = "+requestedNcaID+" ]", EMsgType.PASS);
logPrinter.print(" [Read requested ID = "+requestedNcaID+" ]", EMsgType.PASS);
}
else {
requestedNcaID = pfsElement.getNcaTicketID();
print("GL Handle 'Ticket' command (ID = "+requestedNcaID+" )", EMsgType.INFO);
logPrinter.print("GL Handle 'Ticket' command (ID = "+requestedNcaID+" )", EMsgType.INFO);
}
long realNcaOffset = pfsElement.getNca(requestedNcaID).getNcaOffset()+pfsElement.getBodySize();
@ -317,8 +317,8 @@ public class GoldLeaf_05 extends TransferModule {
logPrinter.updateProgress(1.0);
//-----------------------------------------/
}
catch (IOException | InterruptedException ioe){
print("GL Failed to read NCA ID "+requestedNcaID+". Exception:\n "+ioe.getMessage(), EMsgType.FAIL);
catch (IOException ioe){
logPrinter.print("GL Failed to read NCA ID "+requestedNcaID+". IO Exception:\n "+ioe.getMessage(), EMsgType.FAIL);
ioe.printStackTrace();
return true;
}
@ -345,18 +345,18 @@ public class GoldLeaf_05 extends TransferModule {
if (writeBufTransferred.get() == message.length)
return false;
else {
print("GL Data transfer issue [write]\n Requested: "+message.length+"\n Transferred: "+writeBufTransferred.get(), EMsgType.FAIL);
logPrinter.print("GL Data transfer issue [write]\n Requested: "+message.length+"\n Transferred: "+writeBufTransferred.get(), EMsgType.FAIL);
return true;
}
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [write]\n Returned: "+ UsbErrorCodes.getErrCode(result), EMsgType.FAIL);
print("GL Execution stopped", EMsgType.FAIL);
logPrinter.print("GL Data transfer issue [write]\n Returned: "+ UsbErrorCodes.getErrCode(result), EMsgType.FAIL);
logPrinter.print("GL Execution stopped", EMsgType.FAIL);
return true;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return true;
}
/**
@ -382,12 +382,12 @@ public class GoldLeaf_05 extends TransferModule {
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [read]\n Returned: " + UsbErrorCodes.getErrCode(result), EMsgType.FAIL);
print("GL Execution stopped", EMsgType.FAIL);
logPrinter.print("GL Data transfer issue [read]\n Returned: " + UsbErrorCodes.getErrCode(result), EMsgType.FAIL);
logPrinter.print("GL Execution stopped", EMsgType.FAIL);
return null;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return null;
}
}

View file

@ -69,11 +69,7 @@ class GoldLeaf_07 extends TransferModule {
// For using in CMD_SelectFile with SPEC:/ prefix
private File selectedFile;
GoldLeaf_07(DeviceHandle handler,
LinkedHashMap<String, File> nspMap,
CancellableRunnable task,
ILogPrinter logPrinter,
boolean nspFilter){
GoldLeaf_07(DeviceHandle handler, LinkedHashMap<String, File> nspMap, CancellableRunnable task, ILogPrinter logPrinter, boolean nspFilter){
super(handler, nspMap, task, logPrinter);
final byte CMD_GetDriveCount = 0x00;
@ -97,7 +93,7 @@ class GoldLeaf_07 extends TransferModule {
this.nspFilterForGl = nspFilter;
print("============= GoldLeaf v0.7.x =============\n\t" +
logPrinter.print("============= GoldLeaf v0.7.x =============\n\t" +
"VIRT:/ equals files added into the application\n\t" +
"HOME:/ equals "
+System.getProperty("user.home"), EMsgType.INFO);
@ -264,7 +260,7 @@ class GoldLeaf_07 extends TransferModule {
byte[] drivesCnt = intToArrLE(2); //2
// Write count of drives
if (writeGL_PASS(drivesCnt)) {
print("GL Handle 'ListDrives' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ListDrives' command", EMsgType.FAIL);
return true;
}
return false;
@ -320,7 +316,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(totalSize);
if (writeGL_PASS(command)) {
print("GL Handle 'GetDriveInfo' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDriveInfo' command", EMsgType.FAIL);
return true;
}
@ -336,7 +332,7 @@ class GoldLeaf_07 extends TransferModule {
byte[] specialPathCnt = intToArrLE(0);
// Write count of special paths
if (writeGL_PASS(specialPathCnt)) {
print("GL Handle 'SpecialPathCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'SpecialPathCount' command", EMsgType.FAIL);
return true;
}
return false;
@ -358,13 +354,13 @@ class GoldLeaf_07 extends TransferModule {
if (path.equals("VIRT:/")) {
if (isGetDirectoryCount){
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
return true;
}
}
else {
if (writeGL_PASS(intToArrLE(nspMap.size()))) {
print("GL Handle 'GetFileCount' command Count = "+nspMap.size(), EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFileCount' command Count = "+nspMap.size(), EMsgType.FAIL);
return true;
}
}
@ -405,7 +401,7 @@ class GoldLeaf_07 extends TransferModule {
// If somehow there are no folders, let's say 0;
if (filesOrDirs == null){
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
return true;
}
return false;
@ -419,20 +415,20 @@ class GoldLeaf_07 extends TransferModule {
this.recentFiles = filesOrDirs;
// Otherwise, let's tell how may folders are in there
if (writeGL_PASS(intToArrLE(filesOrDirs.length))) {
print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
return true;
}
}
else if (path.startsWith("SPEC:/")){
if (isGetDirectoryCount){ // If dir request then 0 dirs
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
return true;
}
}
else if (selectedFile != null){ // Else it's file request, if we have selected then we will report 1.
if (writeGL_PASS(intToArrLE(1))) {
print("GL Handle 'GetFileCount' command Count = 1", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFileCount' command Count = 1", EMsgType.FAIL);
return true;
}
}
@ -486,7 +482,7 @@ class GoldLeaf_07 extends TransferModule {
// return proxyGetDirFile(true);
if (writeGL_PASS(command)) {
print("GL Handle 'GetDirectory' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectory' command.", EMsgType.FAIL);
return true;
}
return false;
@ -543,7 +539,7 @@ class GoldLeaf_07 extends TransferModule {
//if (proxyForGL) // TODO: NOTE: PROXY TAILS
// return proxyGetDirFile(false);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -554,7 +550,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(intToArrLE(fileNameBytes.length / 2)); // since GL 0.7
command.add(fileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -566,7 +562,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(intToArrLE(fileNameBytes.length / 2)); // since GL 0.7
command.add(fileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -597,7 +593,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(longToArrLE(fileDirElement.length()));
}
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -614,7 +610,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(longToArrLE(nspMap.get(filePath).length())); // YES, THIS IS LONG!
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -627,7 +623,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(GL_OBJ_TYPE_FILE);
command.add(longToArrLE(selectedFile.length()));
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -655,7 +651,7 @@ class GoldLeaf_07 extends TransferModule {
try {
if (currentFile.renameTo(newFile)){
if (writeGL_PASS()) {
print("GL Handle 'Rename' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Rename' command.", EMsgType.FAIL);
return true;
}
return false;
@ -680,7 +676,7 @@ class GoldLeaf_07 extends TransferModule {
try {
if (fileToDel.delete()){
if (writeGL_PASS()) {
print("GL Handle 'Rename' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Rename' command.", EMsgType.FAIL);
return true;
}
return false;
@ -718,10 +714,10 @@ class GoldLeaf_07 extends TransferModule {
}
if (result){
if (writeGL_PASS()) {
print("GL Handle 'Create' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Create' command.", EMsgType.FAIL);
return true;
}
//print("GL Handle 'Create' command.", EMsgType.PASS);
//logPrinter.print("GL Handle 'Create' command.", EMsgType.PASS);
return false;
}
}
@ -806,12 +802,12 @@ class GoldLeaf_07 extends TransferModule {
"\n Received: " + bytesRead);
// Let's tell as a command about our result.
if (writeGL_PASS(longToArrLE(size))) {
print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
return true;
}
// Let's bypass bytes we read total
if (writeToUsb(chunk)) {
print("GL Handle 'ReadFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -826,12 +822,12 @@ class GoldLeaf_07 extends TransferModule {
return writeGL_FAIL("GL Handle 'ReadFile' command [CMD] Requested = "+size+" Read from file = "+bytesRead);
// Let's tell as a command about our result.
if (writeGL_PASS(longToArrLE(size))) {
print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
return true;
}
// Let's bypass bytes we read total
if (writeToUsb(chunk)) {
print("GL Handle 'ReadFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -843,14 +839,14 @@ class GoldLeaf_07 extends TransferModule {
}
catch (NullPointerException ignored){}
catch (IOException ioe_){
print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
logPrinter.print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
}
try{
splitReader.close();
}
catch (NullPointerException ignored){}
catch (IOException ioe_){
print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
logPrinter.print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
}
openReadFileNameAndPath = null;
randAccessFile = null;
@ -892,7 +888,7 @@ class GoldLeaf_07 extends TransferModule {
byte[] transferredData;
if ((transferredData = readGL_file()) == null){
print("GL Handle 'WriteFile' command [1/1]", EMsgType.FAIL);
logPrinter.print("GL Handle 'WriteFile' command [1/1]", EMsgType.FAIL);
return true;
}
try{
@ -903,7 +899,7 @@ class GoldLeaf_07 extends TransferModule {
}
// Report we're good
if (writeGL_PASS()) {
print("GL Handle 'WriteFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'WriteFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -930,7 +926,7 @@ class GoldLeaf_07 extends TransferModule {
command.add(intToArrLE(selectedFileNameBytes.length / 2)); // since GL 0.7
command.add(selectedFileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'SelectFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'SelectFile' command", EMsgType.FAIL);
this.selectedFile = null;
return true;
}
@ -1010,13 +1006,13 @@ class GoldLeaf_07 extends TransferModule {
closeOpenedReadFilesGl(); // Could be a problem if GL glitches and slow down process. Or if user has extra-slow SD card. TODO: refactor
continue;
default:
print("GL Data transfer issue [read]\n Returned: " +
logPrinter.print("GL Data transfer issue [read]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return null;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return null;
}
private byte[] readGL_file(){
@ -1039,13 +1035,13 @@ class GoldLeaf_07 extends TransferModule {
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [read]\n Returned: " +
logPrinter.print("GL Data transfer issue [read]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return null;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return null;
}
/**
@ -1070,10 +1066,10 @@ class GoldLeaf_07 extends TransferModule {
private boolean writeGL_FAIL(String reportToUImsg){
if (writeToUsb(Arrays.copyOf(CMD_GLCO_FAILURE, 4096))){
print(reportToUImsg, EMsgType.WARNING);
logPrinter.print(reportToUImsg, EMsgType.WARNING);
return true;
}
print(reportToUImsg, EMsgType.FAIL);
logPrinter.print(reportToUImsg, EMsgType.FAIL);
return false;
}
/**
@ -1095,7 +1091,7 @@ class GoldLeaf_07 extends TransferModule {
if (writeBufTransferred.get() == message.length)
return false;
else {
print("GL Data transfer issue [write]\n Requested: " +
logPrinter.print("GL Data transfer issue [write]\n Requested: " +
message.length +
"\n Transferred: " +
writeBufTransferred.get(), EMsgType.FAIL);
@ -1104,13 +1100,13 @@ class GoldLeaf_07 extends TransferModule {
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [write]\n Returned: " +
logPrinter.print("GL Data transfer issue [write]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return true;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return true;
}

View file

@ -103,7 +103,7 @@ class GoldLeaf_08 extends TransferModule {
this.nspFilterForGl = nspFilter;
print("============= GoldLeaf v0.8 =============\n\t" +
logPrinter.print("============= GoldLeaf v0.8 =============\n\t" +
"VIRT:/ equals files added into the application\n\t" +
"HOME:/ equals "
+System.getProperty("user.home"), EMsgType.INFO);
@ -273,7 +273,7 @@ class GoldLeaf_08 extends TransferModule {
* */
private boolean startOrEndFile(){
if (writeGL_PASS()){
print("GL Handle 'StartFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'StartFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -288,7 +288,7 @@ class GoldLeaf_08 extends TransferModule {
byte[] drivesCnt = intToArrLE(2); //2
// Write count of drives
if (writeGL_PASS(drivesCnt)) {
print("GL Handle 'ListDrives' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ListDrives' command", EMsgType.FAIL);
return true;
}
return false;
@ -344,7 +344,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(totalSize);
if (writeGL_PASS(command)) {
print("GL Handle 'GetDriveInfo' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDriveInfo' command", EMsgType.FAIL);
return true;
}
@ -360,7 +360,7 @@ class GoldLeaf_08 extends TransferModule {
byte[] specialPathCnt = intToArrLE(0);
// Write count of special paths
if (writeGL_PASS(specialPathCnt)) {
print("GL Handle 'SpecialPathCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'SpecialPathCount' command", EMsgType.FAIL);
return true;
}
return false;
@ -382,13 +382,13 @@ class GoldLeaf_08 extends TransferModule {
if (path.equals("VIRT:/")) {
if (isGetDirectoryCount){
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
return true;
}
}
else {
if (writeGL_PASS(intToArrLE(nspMap.size()))) {
print("GL Handle 'GetFileCount' command Count = "+nspMap.size(), EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFileCount' command Count = "+nspMap.size(), EMsgType.FAIL);
return true;
}
}
@ -429,7 +429,7 @@ class GoldLeaf_08 extends TransferModule {
// If somehow there are no folders, let's say 0;
if (filesOrDirs == null){
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
return true;
}
return false;
@ -443,20 +443,20 @@ class GoldLeaf_08 extends TransferModule {
this.recentFiles = filesOrDirs;
// Otherwise, let's tell how may folders are in there
if (writeGL_PASS(intToArrLE(filesOrDirs.length))) {
print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryOrFileCount' command", EMsgType.FAIL);
return true;
}
}
else if (path.startsWith("SPEC:/")){
if (isGetDirectoryCount){ // If dir request then 0 dirs
if (writeGL_PASS()) {
print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectoryCount' command", EMsgType.FAIL);
return true;
}
}
else if (selectedFile != null){ // Else it's file request, if we have selected then we will report 1.
if (writeGL_PASS(intToArrLE(1))) {
print("GL Handle 'GetFileCount' command Count = 1", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFileCount' command Count = 1", EMsgType.FAIL);
return true;
}
}
@ -510,7 +510,7 @@ class GoldLeaf_08 extends TransferModule {
// return proxyGetDirFile(true);
if (writeGL_PASS(command)) {
print("GL Handle 'GetDirectory' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetDirectory' command.", EMsgType.FAIL);
return true;
}
return false;
@ -567,7 +567,7 @@ class GoldLeaf_08 extends TransferModule {
//if (proxyForGL) // TODO: NOTE: PROXY TAILS
// return proxyGetDirFile(false);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -578,7 +578,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(intToArrLE(fileNameBytes.length / 2)); // since GL 0.7
command.add(fileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -590,7 +590,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(intToArrLE(fileNameBytes.length / 2)); // since GL 0.7
command.add(fileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'GetFile' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'GetFile' command.", EMsgType.FAIL);
return true;
}
return false;
@ -621,7 +621,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(longToArrLE(fileDirElement.length()));
}
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -638,7 +638,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(longToArrLE(nspMap.get(filePath).length())); // YES, THIS IS LONG!
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -651,7 +651,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(GL_OBJ_TYPE_FILE);
command.add(longToArrLE(selectedFile.length()));
if (writeGL_PASS(command)) {
print("GL Handle 'StatPath' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'StatPath' command.", EMsgType.FAIL);
return true;
}
return false;
@ -679,7 +679,7 @@ class GoldLeaf_08 extends TransferModule {
try {
if (currentFile.renameTo(newFile)){
if (writeGL_PASS()) {
print("GL Handle 'Rename' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Rename' command.", EMsgType.FAIL);
return true;
}
return false;
@ -704,7 +704,7 @@ class GoldLeaf_08 extends TransferModule {
try {
if (fileToDel.delete()){
if (writeGL_PASS()) {
print("GL Handle 'Rename' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Rename' command.", EMsgType.FAIL);
return true;
}
return false;
@ -742,10 +742,10 @@ class GoldLeaf_08 extends TransferModule {
}
if (result){
if (writeGL_PASS()) {
print("GL Handle 'Create' command.", EMsgType.FAIL);
logPrinter.print("GL Handle 'Create' command.", EMsgType.FAIL);
return true;
}
//print("GL Handle 'Create' command.", EMsgType.PASS);
//logPrinter.print("GL Handle 'Create' command.", EMsgType.PASS);
return false;
}
}
@ -830,12 +830,12 @@ class GoldLeaf_08 extends TransferModule {
"\n Received: " + bytesRead);
// Let's tell as a command about our result.
if (writeGL_PASS(longToArrLE(size))) {
print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
return true;
}
// Let's bypass bytes we read total
if (writeToUsb(chunk)) {
print("GL Handle 'ReadFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -850,12 +850,12 @@ class GoldLeaf_08 extends TransferModule {
return writeGL_FAIL("GL Handle 'ReadFile' command [CMD] Requested = "+size+" Read from file = "+bytesRead);
// Let's tell as a command about our result.
if (writeGL_PASS(longToArrLE(size))) {
print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command [CMD]", EMsgType.FAIL);
return true;
}
// Let's bypass bytes we read total
if (writeToUsb(chunk)) {
print("GL Handle 'ReadFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'ReadFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -867,14 +867,14 @@ class GoldLeaf_08 extends TransferModule {
}
catch (NullPointerException ignored){}
catch (IOException ioe_){
print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
logPrinter.print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
}
try{
splitReader.close();
}
catch (NullPointerException ignored){}
catch (IOException ioe_){
print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
logPrinter.print("GL Handle 'ReadFile' command: unable to close: "+openReadFileNameAndPath+"\n\t"+ioe_.getMessage(), EMsgType.WARNING);
}
openReadFileNameAndPath = null;
randAccessFile = null;
@ -916,7 +916,7 @@ class GoldLeaf_08 extends TransferModule {
byte[] transferredData;
if ((transferredData = readGL_file()) == null){
print("GL Handle 'WriteFile' command [1/1]", EMsgType.FAIL);
logPrinter.print("GL Handle 'WriteFile' command [1/1]", EMsgType.FAIL);
return true;
}
try{
@ -927,7 +927,7 @@ class GoldLeaf_08 extends TransferModule {
}
// Report we're good
if (writeGL_PASS()) {
print("GL Handle 'WriteFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'WriteFile' command", EMsgType.FAIL);
return true;
}
return false;
@ -957,7 +957,7 @@ class GoldLeaf_08 extends TransferModule {
command.add(intToArrLE(selectedFileNameBytes.length / 2)); // since GL 0.7
command.add(selectedFileNameBytes);
if (writeGL_PASS(command)) {
print("GL Handle 'SelectFile' command", EMsgType.FAIL);
logPrinter.print("GL Handle 'SelectFile' command", EMsgType.FAIL);
this.selectedFile = null;
return true;
}
@ -1033,13 +1033,13 @@ class GoldLeaf_08 extends TransferModule {
closeOpenedReadFilesGl(); // Could be a problem if GL glitches and slow down process. Or if user has extra-slow SD card. TODO: refactor
continue;
default:
print("GL Data transfer issue [read]\n Returned: " +
logPrinter.print("GL Data transfer issue [read]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return null;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return null;
}
private byte[] readGL_file(){
@ -1060,13 +1060,13 @@ class GoldLeaf_08 extends TransferModule {
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [read]\n Returned: " +
logPrinter.print("GL Data transfer issue [read]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return null;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return null;
}
/**
@ -1091,10 +1091,10 @@ class GoldLeaf_08 extends TransferModule {
private boolean writeGL_FAIL(String reportToUImsg){
if (writeToUsb(Arrays.copyOf(CMD_GLCO_FAILURE, 4096))){
print(reportToUImsg, EMsgType.WARNING);
logPrinter.print(reportToUImsg, EMsgType.WARNING);
return true;
}
print(reportToUImsg, EMsgType.FAIL);
logPrinter.print(reportToUImsg, EMsgType.FAIL);
return false;
}
/**
@ -1116,7 +1116,7 @@ class GoldLeaf_08 extends TransferModule {
if (writeBufTransferred.get() == message.length)
return false;
else {
print("GL Data transfer issue [write]\n Requested: " +
logPrinter.print("GL Data transfer issue [write]\n Requested: " +
message.length +
"\n Transferred: "+writeBufTransferred.get(), EMsgType.FAIL);
return true;
@ -1124,13 +1124,13 @@ class GoldLeaf_08 extends TransferModule {
case LibUsb.ERROR_TIMEOUT:
continue;
default:
print("GL Data transfer issue [write]\n Returned: " +
logPrinter.print("GL Data transfer issue [write]\n Returned: " +
UsbErrorCodes.getErrCode(result) +
"\n GL Execution stopped", EMsgType.FAIL);
return true;
}
}
print("GL Execution interrupted", EMsgType.INFO);
logPrinter.print("GL Execution interrupted", EMsgType.INFO);
return true;
}
}

View file

@ -50,7 +50,7 @@ class TinFoil extends TransferModule {
TinFoil(DeviceHandle handler, LinkedHashMap<String, File> nspMap, CancellableRunnable task, ILogPrinter logPrinter){
super(handler, nspMap, task, logPrinter);
print("============= Tinfoil =============", EMsgType.INFO);
logPrinter.print("============= Tinfoil =============", EMsgType.INFO);
if (! sendListOfFiles())
return;
@ -69,25 +69,25 @@ class TinFoil extends TransferModule {
byte[] padding = new byte[8];
if (writeUsb(TUL0)) {
print("TF Send list of files: handshake [1/4]", EMsgType.FAIL);
logPrinter.print("TF Send list of files: handshake [1/4]", EMsgType.FAIL);
return false;
}
if (writeUsb(nspListNamesSize)) { // size of the list we can transfer
print("TF Send list of files: list length [2/4]", EMsgType.FAIL);
logPrinter.print("TF Send list of files: list length [2/4]", EMsgType.FAIL);
return false;
}
if (writeUsb(padding)) {
print("TF Send list of files: padding [3/4]", EMsgType.FAIL);
logPrinter.print("TF Send list of files: padding [3/4]", EMsgType.FAIL);
return false;
}
if (writeUsb(nspListNames)) {
print("TF Send list of files: list itself [4/4]", EMsgType.FAIL);
logPrinter.print("TF Send list of files: list itself [4/4]", EMsgType.FAIL);
return false;
}
print("TF Send list of files complete.", EMsgType.PASS);
logPrinter.print("TF Send list of files complete.", EMsgType.PASS);
return true;
}
@ -114,7 +114,7 @@ class TinFoil extends TransferModule {
* After we sent commands to NS, this chain starts
* */
private boolean proceedCommands(){
print("TF Awaiting for NS commands.", EMsgType.INFO);
logPrinter.print("TF Awaiting for NS commands.", EMsgType.INFO);
try{
byte[] deviceReply;
byte command;
@ -127,18 +127,18 @@ class TinFoil extends TransferModule {
switch (command){
case CMD_EXIT:
print("TF Transfer complete.", EMsgType.PASS);
logPrinter.print("TF Transfer complete.", EMsgType.PASS);
return true;
case CMD_FILE_RANGE_DEFAULT:
case CMD_FILE_RANGE_ALTERNATIVE:
//print("TF Received 'FILE RANGE' command [0x0"+command+"].", EMsgType.PASS);
//logPrinter.print("TF Received 'FILE RANGE' command [0x0"+command+"].", EMsgType.PASS);
if (fileRangeCmd())
return false; // catches exception
}
}
}
catch (Exception e){
print(e.getMessage(), EMsgType.INFO);
logPrinter.print(e.getMessage(), EMsgType.INFO);
return false;
}
}
@ -169,7 +169,7 @@ class TinFoil extends TransferModule {
String nspFileName = new String(receivedArray, StandardCharsets.UTF_8);
print(String.format("TF Reply to: %s" +
logPrinter.print(String.format("TF Reply to: %s" +
"\n Offset: %-20d 0x%x" +
"\n Size: %-20d 0x%x",
nspFileName,
@ -188,28 +188,28 @@ class TinFoil extends TransferModule {
else
sendNormalFile(nspFile, size, offset);
} catch (IOException ioe){
print("TF IOException:\n "+ioe.getMessage(), EMsgType.FAIL);
logPrinter.print("TF IOException:\n "+ioe.getMessage(), EMsgType.FAIL);
ioe.printStackTrace();
return true;
} catch (ArithmeticException ae){
print("TF ArithmeticException (can't cast 'offset end' - 'offsets current' to 'integer'):" +
logPrinter.print("TF ArithmeticException (can't cast 'offset end' - 'offsets current' to 'integer'):" +
"\n "+ae.getMessage(), EMsgType.FAIL);
ae.printStackTrace();
return true;
} catch (NullPointerException npe){
print("TF NullPointerException (in some moment application didn't find something. Something important.):" +
logPrinter.print("TF NullPointerException (in some moment application didn't find something. Something important.):" +
"\n "+npe.getMessage(), EMsgType.FAIL);
npe.printStackTrace();
return true;
}
catch (Exception defe){
print(defe.getMessage(), EMsgType.FAIL);
logPrinter.print(defe.getMessage(), EMsgType.FAIL);
return true;
}
return false;
}
void sendSplitFile(File nspFile, long size, long offset) throws Exception {
void sendSplitFile(File nspFile, long size, long offset) throws IOException, NullPointerException, ArithmeticException {
byte[] readBuffer;
long currentOffset = 0;
int chunk = 8388608; // = 8Mb;
@ -237,7 +237,7 @@ class TinFoil extends TransferModule {
logPrinter.updateProgress(1.0);
}
void sendNormalFile(File nspFile, long size, long offset) throws Exception {
void sendNormalFile(File nspFile, long size, long offset) throws IOException, NullPointerException, ArithmeticException {
byte[] readBuffer;
long currentOffset = 0;
int chunk = 8388608;
@ -278,17 +278,17 @@ class TinFoil extends TransferModule {
final byte[] twelveZeroBytes = new byte[12];
if (writeUsb(standardReplyBytes)){ // Send integer value of '1' in Little-endian format.
print("TF Sending response failed [1/3]", EMsgType.FAIL);
logPrinter.print("TF Sending response failed [1/3]", EMsgType.FAIL);
return true;
}
if(writeUsb(sizeAsBytes)) { // Send EXACTLY what has been received
print("TF Sending response failed [2/3]", EMsgType.FAIL);
logPrinter.print("TF Sending response failed [2/3]", EMsgType.FAIL);
return true;
}
if(writeUsb(twelveZeroBytes)) { // kinda another one padding
print("TF Sending response failed [3/3]", EMsgType.FAIL);
logPrinter.print("TF Sending response failed [3/3]", EMsgType.FAIL);
return true;
}
return false;
@ -308,7 +308,7 @@ class TinFoil extends TransferModule {
while (! task.isCancelled() ) {
/*
if (varVar != 0)
print("writeUsb() retry cnt: "+varVar, EMsgType.INFO); //NOTE: DEBUG
logPrinter.print("writeUsb() retry cnt: "+varVar, EMsgType.INFO); //NOTE: DEBUG
varVar++;
*/
result = LibUsb.bulkTransfer(handlerNS, (byte) 0x01, writeBuffer, writeBufTransferred, 5050); // last one is TIMEOUT. 0 stands for unlimited. Endpoint OUT = 0x01
@ -317,7 +317,7 @@ class TinFoil extends TransferModule {
case LibUsb.SUCCESS:
if (writeBufTransferred.get() == message.length)
return false;
print("TF Data transfer issue [write]" +
logPrinter.print("TF Data transfer issue [write]" +
"\n Requested: "+message.length+
"\n Transferred: "+writeBufTransferred.get(), EMsgType.FAIL);
return true;
@ -326,13 +326,13 @@ class TinFoil extends TransferModule {
//writeBufTransferred.clear(); // MUST BE HERE IF WE 'GET()' IT
continue;
default:
print("TF Data transfer issue [write]" +
logPrinter.print("TF Data transfer issue [write]" +
"\n Returned: "+ UsbErrorCodes.getErrCode(result) +
"\n (execution stopped)", EMsgType.FAIL);
return true;
}
}
print("TF Execution interrupted", EMsgType.INFO);
logPrinter.print("TF Execution interrupted", EMsgType.INFO);
return true;
}
/**

View file

@ -51,7 +51,7 @@ public abstract class TransferModule {
File[] subFiles = f.listFiles((file, name) -> name.matches("[0-9]{2}"));
if (subFiles == null || subFiles.length == 0) {
print("TransferModule: Exclude folder: " + f.getName(), EMsgType.WARNING);
logPrinter.print("TransferModule: Exclude folder: " + f.getName(), EMsgType.WARNING);
return true;
}
@ -59,7 +59,7 @@ public abstract class TransferModule {
for (int i = subFiles.length - 2; i > 0 ; i--){
if (subFiles[i].length() != subFiles[i-1].length()) {
print("TransferModule: Exclude split file: "+f.getName()+
logPrinter.print("TransferModule: Exclude split file: "+f.getName()+
"\n Chunk sizes of the split file are not the same, but has to be.", EMsgType.WARNING);
return true;
}
@ -69,7 +69,7 @@ public abstract class TransferModule {
long lastFileLength = subFiles[subFiles.length-1].length();
if (lastFileLength > firstFileLength){
print("TransferModule: Exclude split file: "+f.getName()+
logPrinter.print("TransferModule: Exclude split file: "+f.getName()+
"\n Chunk sizes of the split file are not the same, but has to be.", EMsgType.WARNING);
return true;
}
@ -77,13 +77,4 @@ public abstract class TransferModule {
});
}
public EFileStatus getStatus(){ return status; }
void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -49,7 +49,7 @@ public class UsbCommunications extends CancellableRunnable {
@Override
public void run() {
print("\tStart", EMsgType.INFO);
logPrinter.print("\tStart", EMsgType.INFO);
UsbConnect usbConnect = UsbConnect.connectHomebrewMode(logPrinter);
@ -87,15 +87,7 @@ public class UsbCommunications extends CancellableRunnable {
*/
private void close(EFileStatus status){
logPrinter.update(nspMap, status);
print("\tEnd", EMsgType.INFO);
logPrinter.print("\tEnd", EMsgType.INFO);
logPrinter.close();
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -65,10 +65,7 @@ public class UsbConnect {
usbConnect.connected = true;
}
catch (Exception e){
try {
logPrinter.print(e.getMessage(), EMsgType.FAIL);
}
catch (InterruptedException ignore){}
usbConnect.close();
}
@ -92,10 +89,7 @@ public class UsbConnect {
}
catch (Exception e){
e.printStackTrace();
try {
logPrinter.print(e.getMessage(), EMsgType.FAIL);
}
catch (InterruptedException ignore){}
usbConnect.close();
}
return usbConnect;
@ -106,7 +100,7 @@ public class UsbConnect {
private UsbConnect(ILogPrinter logPrinter){
this.logPrinter = logPrinter;
this.connected = false;
}
};
private void createContextAndInitLibUSB() throws Exception{
// Creating Context required by libusb. Optional? Consider removing.
@ -184,7 +178,7 @@ public class UsbConnect {
// Actually, there are no drivers in Linux kernel which uses this device.
returningValue = LibUsb.setAutoDetachKernelDriver(handlerNS, true);
if (returningValue != LibUsb.SUCCESS)
print("Skip kernel driver attach & detach ("+UsbErrorCodes.getErrCode(returningValue)+")", EMsgType.INFO);
logPrinter.print("Skip kernel driver attach & detach ("+UsbErrorCodes.getErrCode(returningValue)+")", EMsgType.INFO);
}
/*
@ -236,7 +230,7 @@ public class UsbConnect {
returningValue = LibUsb.releaseInterface(handlerNS, DEFAULT_INTERFACE);
if (returningValue != LibUsb.SUCCESS) {
print("Release interface failure: " +
logPrinter.print("Release interface failure: " +
UsbErrorCodes.getErrCode(returningValue) +
" (sometimes it's not an issue)", EMsgType.WARNING);
}
@ -247,13 +241,4 @@ public class UsbConnect {
if (contextNS != null)
LibUsb.exit(contextNS);
}
private void print(String message, EMsgType type){
try {
logPrinter.print(message, type);
}
catch (InterruptedException ie){
ie.printStackTrace();
}
}
}

View file

@ -1,10 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<?import javafx.scene.control.ListView?>
<?import javafx.scene.layout.AnchorPane?>
<AnchorPane xmlns="http://javafx.com/javafx/8.0.141" xmlns:fx="http://javafx.com/fxml/1" fx:controller="nsusbloader.Controllers.BlockListViewController">
<children>
<ListView fx:id="splitMergeListView" AnchorPane.bottomAnchor="0.0" AnchorPane.leftAnchor="0.0" AnchorPane.rightAnchor="0.0" AnchorPane.topAnchor="0.0" />
</children>
</AnchorPane>

View file

@ -42,10 +42,7 @@
</GridPane>
</children>
</VBox>
<HBox spacing="15.0" VBox.vgrow="ALWAYS">
<children>
<fx:include fx:id="BlockListView" source="BlockListView.fxml" HBox.hgrow="ALWAYS" VBox.vgrow="ALWAYS" />
<VBox spacing="5.0">
<VBox fillWidth="false" spacing="5.0">
<children>
<RadioButton fx:id="splitRad" contentDisplay="TOP" mnemonicParsing="false" text="%tabSplMrg_RadioBtn_Split">
<toggleGroup>
@ -53,14 +50,47 @@
</toggleGroup>
</RadioButton>
<RadioButton fx:id="mergeRad" contentDisplay="TOP" mnemonicParsing="false" text="%tabSplMrg_RadioBtn_Merge" toggleGroup="$splitMergeTogGrp" />
<Button fx:id="selectFileFolderBtn" contentDisplay="TOP" mnemonicParsing="false">
</children>
<VBox.margin>
<Insets bottom="10.0" />
<Insets left="15.0" right="15.0" />
</VBox.margin>
</Button>
</VBox>
<VBox spacing="5.0">
<children>
<HBox>
<children>
<Label fx:id="fileFolderLabelLbl" />
<Label fx:id="fileFolderActualPathLbl" />
</children>
</HBox>
<Button fx:id="selectFileFolderBtn" contentDisplay="TOP" mnemonicParsing="false" />
</children>
<VBox.margin>
<Insets left="15.0" right="15.0" />
</VBox.margin>
</VBox>
<VBox spacing="5.0">
<children>
<HBox>
<children>
<Label text="%tabSplMrg_Lbl_SaveToLocation" />
<Label fx:id="saveToPathLbl" maxWidth="200.0" textOverrun="CENTER_WORD_ELLIPSIS" />
<Label fx:id="saveToPathLbl" />
</children>
</HBox>
<Button fx:id="changeSaveToBtn" contentDisplay="TOP" mnemonicParsing="false" text="%tabSplMrg_Btn_ChangeSaveToLocation" />
</children>
<VBox.margin>
<Insets left="15.0" right="15.0" />
</VBox.margin>
</VBox>
<HBox alignment="CENTER">
<children>
<Label fx:id="statusLbl" />
</children>
<VBox.margin>
<Insets left="15.0" right="15.0" />
</VBox.margin>
</HBox>
<Pane VBox.vgrow="ALWAYS" />
<VBox>
<children>
@ -71,21 +101,6 @@
</HBox>
</children>
</VBox>
</children>
</VBox>
</children>
<VBox.margin>
<Insets left="15.0" right="15.0" />
</VBox.margin>
</HBox>
<HBox alignment="CENTER">
<children>
<Label fx:id="statusLbl" />
</children>
<VBox.margin>
<Insets left="15.0" right="15.0" />
</VBox.margin>
</HBox>
<padding>
<Insets bottom="5.0" left="5.0" right="5.0" top="5.0" />
</padding>

View file

@ -205,7 +205,6 @@
.table-view .table-cell{
-fx-text-fill: #f7fafa;
}
.table-row-cell, .table-row-cell:selected, .table-row-cell:filled:selected{
-fx-background-color: -fx-table-cell-border-color, #424242;
-fx-background-insets: 0, 0 0 1 0;
@ -224,35 +223,7 @@
-fx-padding: 0.0em; /* 0 */
-fx-table-cell-border-color: #6d8484;
}
/* -========================== ListView =====================- */
.list-view {
-fx-background-color: #4f4f4f;
-fx-background-position: center;
-fx-background-repeat: no-repeat;
-fx-background-radius: 3;
-fx-border-color: #00ffc9;
-fx-border-radius: 3;
-fx-border-width: 2;
}
.list-cell, .list-cell:selected, .list-cell:filled:selected{
-fx-background-color: -fx-table-cell-border-color, #424242;
-fx-background-insets: 0, 0 0 1 0;
-fx-table-cell-border-color: #6d8484;
}
.list-cell:odd, .list-cell:odd:selected, .list-cell:odd:filled:selected{
-fx-background-color: -fx-table-cell-border-color, #4f4f4f;
-fx-background-insets: 0, 0 0 1 0;
-fx-table-cell-border-color: #6d8484;
}
.list-cell .text, .list-cell:odd .text{
-fx-fill: #f7fafa;
}
.list-cell:filled:selected .text, .list-cell:odd:filled:selected .text{
-fx-fill: #08f3ff;
}
/* -========================== Context menu =====================- */
// -========================== Context menu =====================-
.context-menu {
-fx-background-color: #2d2d2d;
-fx-cursor: hand;
@ -263,7 +234,7 @@
.context-menu .menu-item:focused .label {
-fx-text-fill: white;
}
/* -========================== Text Field =====================- */
// -========================== Text Field =====================-
.text-field {
-fx-prompt-text-fill: #40596c;
-fx-border-color: #289de8;

View file

@ -242,35 +242,6 @@
-fx-padding: 0.0em; /* 0 */
-fx-table-cell-border-color: #b0b0b0;
}
/* -========================== ListView =====================- */
.list-view {
-fx-background-color: #fefefe;
-fx-background-position: center;
-fx-background-repeat: no-repeat;
-fx-background-radius: 3;
-fx-border-color: #06b9bb;
-fx-border-radius: 3;
-fx-border-width: 2;
}
.list-cell, .list-cell:selected, .list-cell:filled:selected{
-fx-background-color: -fx-table-cell-border-color, #ebfffe;
-fx-background-insets: 0, 0 0 1 0;
-fx-table-cell-border-color: #b0b0b0;
}
.list-cell:odd, .list-cell:odd:selected, .list-cell:odd:filled:selected{
-fx-background-color: -fx-table-cell-border-color, #fefefe;
-fx-background-insets: 0, 0 0 1 0;
-fx-table-cell-border-color: #b0b0b0;
}
.list-cell .text, .list-cell:odd .text{
-fx-fill: #2c2c2c;
}
.list-cell:filled:selected .text, .list-cell:odd:filled:selected .text{
-fx-fill: #2c2c2c;
-fx-font-weight: bold
}
/* -========================= Separator ===================- */
.separator *.line {
-fx-border-style: solid;

View file

@ -1,86 +0,0 @@
package nsusbloader.com.usb;
import nsusbloader.NSLMain;
import nsusbloader.Utilities.splitmerge.SplitMergeTaskExecutor;
import org.junit.jupiter.api.*;
import org.junit.jupiter.api.io.TempDir;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
public class MergeTest {
private List<File> files;
@TempDir
File testFilesLocation;
@TempDir
File saveToPath;
static Random random;
@BeforeAll
static void init(){
NSLMain.isCli = true;
MergeTest.random = new Random();
}
void makeSplittedFiles() throws Exception{
String parentLocation = testFilesLocation.getAbsolutePath();
this.files = new ArrayList<>();
for (int i = 0; i < 7; i++) {
files.add(createSplitFile(parentLocation, "TestFile4Merge_" + i));
}
}
File createSplitFile(String parent, String name) throws Exception{
Path file = Paths.get(parent, name);
File newlyCreatedSplitFile = Files.createDirectory(file).toFile();
Assertions.assertTrue(newlyCreatedSplitFile.exists());
Assertions.assertTrue(newlyCreatedSplitFile.canRead());
Assertions.assertTrue(newlyCreatedSplitFile.canWrite());
int chunksCount = random.nextInt(6 + 1) + 1; // At min = 1, max = 6
populateSplittedFile(newlyCreatedSplitFile, chunksCount);
return newlyCreatedSplitFile;
}
void populateSplittedFile(File splitFileContainer, int chunksCount) throws Exception{
int chunkSize = random.nextInt(8192 + 1) + 8192; // At min = 8192, max = 8192*2
for (int i = 0; i < chunksCount; i++){
File chunkFile = new File(splitFileContainer, String.format("%02d", i));
try (BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(chunkFile))){
byte[] zero = new byte[chunkSize];
bos.write(zero);
}
Assertions.assertTrue(chunkFile.exists());
Assertions.assertTrue(chunkFile.canRead());
}
}
@DisplayName("Test test-files location for merge")
@Test
void testTempLocation(){
Assertions.assertTrue(testFilesLocation.isDirectory());
}
// @Disabled("Current test is not ready")
@DisplayName("Test merge functionality")
//@Timeout(value = 60000, unit = TimeUnit.MILLISECONDS)
@Test
void testMerge() throws Exception{
makeSplittedFiles();
SplitMergeTaskExecutor splitMergeTaskExecutor = new SplitMergeTaskExecutor(false, files, saveToPath.getAbsolutePath());
Thread thread = new Thread(splitMergeTaskExecutor);
thread.setDaemon(true);
thread.start();
thread.join();
}
}

View file

@ -1,78 +0,0 @@
package nsusbloader.com.usb;
import nsusbloader.NSLMain;
import nsusbloader.Utilities.splitmerge.SplitMergeTaskExecutor;
import org.junit.jupiter.api.*;
import org.junit.jupiter.api.io.TempDir;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
class SplitTest {
private List<File> files;
@TempDir
File testFilesLocation;
@TempDir
File saveToPath;
static Random random;
@BeforeAll
static void init(){
NSLMain.isCli = true;
SplitTest.random = new Random();
}
void makeRegularFiles() throws Exception {
String parentLocation = testFilesLocation.getAbsolutePath();
Assertions.assertTrue(Files.exists(Paths.get(parentLocation)));
this.files = new ArrayList<>();
for (int i = 0; i < 7; i++) {
files.add(createRegularFile(parentLocation, "TestFile4Split_" + i));
}
}
File createRegularFile(String parent, String name) throws Exception{
Path file = Paths.get(parent, name);
File newlyCreatedFile = Files.createFile(file).toFile();
Assertions.assertTrue(newlyCreatedFile.exists());
Assertions.assertTrue(newlyCreatedFile.canRead());
Assertions.assertTrue(newlyCreatedFile.canWrite());
int randomValue = random.nextInt(8192 + 1) + 8192; // At min = 8192, max = 8192*2
fulfillFile(newlyCreatedFile, randomValue);
return newlyCreatedFile;
}
void fulfillFile(File file, int fileSize) throws Exception{
try (BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(file))){
byte[] zero = new byte[fileSize];
bos.write(zero);
}
}
@DisplayName("Test test-files location for split")
@Test
void testTempLocation(){
Assertions.assertTrue(testFilesLocation.isDirectory());
}
@DisplayName("Test split functionality")
//@Timeout(value = 60000, unit = TimeUnit.MILLISECONDS)
@Test
void testSplit() throws Exception{
makeRegularFiles();
SplitMergeTaskExecutor splitMergeTaskExecutor = new SplitMergeTaskExecutor(true, files, saveToPath.getAbsolutePath());
Thread thread = new Thread(splitMergeTaskExecutor);
thread.setDaemon(true);
thread.start();
thread.join();
}
}