diff --git a/.gitignore b/.gitignore index 524f096..2d00de4 100644 --- a/.gitignore +++ b/.gitignore @@ -22,3 +22,9 @@ # virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml hs_err_pid* replay_pid* + +# IDE Stuff +.vscode/* +.vscode/ +*.vscode +.vscode \ No newline at end of file diff --git a/BlockSize.java b/BlockSize.java new file mode 100644 index 0000000..02bab44 --- /dev/null +++ b/BlockSize.java @@ -0,0 +1,38 @@ +import java.util.Scanner; + +public class BlockSize { + + public static String blockSizeString = ""; + + static String blockSize(Scanner UserInput) { + System.out.println("Choose a block size (Default: 4M)"); + System.out.println("512KB (1), 1M (2), 2M (3), 4M (4), 8M (5), 16M (6)"); + String blockSizeInput = UserInput.nextLine(); + + switch (blockSizeInput) { + case "1": + blockSizeString = "512KB"; + break; + case "2": + blockSizeString = "1M"; + break; + case "3": + blockSizeString = "2M"; + break; + case "4": + blockSizeString = "4M"; + break; + case "5": + blockSizeString = "8M"; + break; + case "6": + blockSizeString = "16M"; + break; + default: + blockSizeString = "4M"; + break; + } + System.out.println("Using blocksize of: " + blockSizeString); + return blockSizeString; + } +} diff --git a/Dd.java b/Dd.java new file mode 100644 index 0000000..9db9a6a --- /dev/null +++ b/Dd.java @@ -0,0 +1,26 @@ +import java.io.BufferedReader; +import java.io.IOException; +import java.io.InputStreamReader; + +public class Dd { + public static void dd() { + try { + ProcessBuilder pb = new ProcessBuilder("sudo", "dd", "if=" + FilePathAdd.ImagePath, "of=" + StorageDeviceLister.fullPath, "bs=" + BlockSize.blockSizeString, "status=progress", "oflag=" + OflagHandler.oflagHandleString); + pb.redirectErrorStream(true); + Process process = pb.start(); + + BufferedReader reader = new BufferedReader( + new InputStreamReader(process.getInputStream()) + ); + + int character; + while ((character = reader.read()) != -1) { + char c = (char) character; + System.out.print(c); + System.out.flush(); + } + } catch (IOException e) { + e.printStackTrace(); + } + } +} diff --git a/FilePathAdd.java b/FilePathAdd.java new file mode 100644 index 0000000..9176d3b --- /dev/null +++ b/FilePathAdd.java @@ -0,0 +1,13 @@ +import java.util.Scanner; + +public class FilePathAdd { + + public static String ImagePath = ""; + + protected static String filePath(Scanner UserInput) { + System.out.println("Please enter the FULL Path of your ISO / Image. ()"); + ImagePath = UserInput.nextLine(); + System.out.println("Using File: " + ImagePath); + return ImagePath; + } +} diff --git a/Flasher.java b/Flasher.java new file mode 100644 index 0000000..3cca041 --- /dev/null +++ b/Flasher.java @@ -0,0 +1,23 @@ +import java.util.Scanner; + +public class Flasher { + public static void flasher(Scanner UserInput) { + + String input = ""; + + System.out.println("The programm wil use the following configuration, do you want to flash with this? (y/n)"); + System.out.println(" - To be flashed device: " + StorageDeviceLister.fullPath); + System.out.println(" - To be used path: " + FilePathAdd.ImagePath); + System.out.println(" - To bed used blocksize: " + BlockSize.blockSizeString); + System.out.println(" - To be used oflag: " + OflagHandler.oflagHandleString); + input = UserInput.nextLine(); + if (YesNo.check(input)) { + System.out.println("Starting to flash..."); + } else { + System.out.println("Canceling..."); + System.exit(0); + } + Dd.dd(); + System.out.println("Flash completed."); + } +} diff --git a/Greeting.java b/Greeting.java new file mode 100644 index 0000000..5c286c2 --- /dev/null +++ b/Greeting.java @@ -0,0 +1,6 @@ +public class Greeting { + public static void greeting() { + System.out.println("Welcome to cflash!"); + System.out.println("Would you like to flash an image (Y/n)"); + } +} diff --git a/Main.java b/Main.java new file mode 100644 index 0000000..58e37b8 --- /dev/null +++ b/Main.java @@ -0,0 +1,24 @@ +import java.util.Scanner; + +public class Main { + public static void main(String[] args) throws InterruptedException { + Scanner UserInput = new Scanner(System.in); + + Greeting.greeting(); + String input = UserInput.nextLine(); + + if (YesNo.check(input)) { + System.out.println("Please choose the to be flashed device (f. e. sda)"); + } else { + System.out.println("Canceling..."); + System.exit(0); + } + StorageDeviceLister.deviceList(UserInput); + FilePathAdd.filePath(UserInput); + BlockSize.blockSize(UserInput); + OflagHandler.handleOflag(UserInput); + Flasher.flasher(UserInput); + + + } +} diff --git a/OflagHandler.java b/OflagHandler.java new file mode 100644 index 0000000..2fa4016 --- /dev/null +++ b/OflagHandler.java @@ -0,0 +1,32 @@ +import java.util.Scanner; + +public class OflagHandler { + + public static String oflagHandleString = ""; + + static String handleOflag(Scanner UserInput) { + System.out.println("Okay, next up please define your oflag (Default: direct)"); + System.out.println("Available flags: direct (1), dsync (2), sync (3), nocache (4)"); + String oflagHandleInput = UserInput.nextLine(); + + switch (oflagHandleInput) { + case "1": + oflagHandleString = "direct"; + break; + case "2": + oflagHandleString = "dsync"; + break; + case "3": + oflagHandleString = "sync"; + break; + case "4": + oflagHandleString = "nocache"; + break; + default: + oflagHandleString = "direct"; + break; + } + System.out.println("Using oflag: " + oflagHandleString); + return oflagHandleString; + } +} diff --git a/README.md b/README.md index 2d3500b..d1ac985 100644 --- a/README.md +++ b/README.md @@ -7,7 +7,7 @@ Small and lightweight image and iso flasher build on `dd`. [![Status](https://img.shields.io/badge/Status-Beta-red.svg)](https://en.wikipedia.org/wiki/Software_release_life_cycle#Beta) # About -Java program using `dd` to make flashing iso and image files easier on the terminal. This program allows anyone to flash iso and image files without having to search for extra GUI tools and by keeping it simple and resource-friendly. +Java program using `dd` to make flashing iso and image files easier on the terminal. This program allows anyone to flash iso and image files without having to search for extra GUI tools by keeping it simple and resource-friendly. # Requirements - `Java`: 21 (Download [here](https://www.oracle.com/java/technologies/downloads/#java21) diff --git a/StorageDeviceLister.java b/StorageDeviceLister.java new file mode 100644 index 0000000..196a92f --- /dev/null +++ b/StorageDeviceLister.java @@ -0,0 +1,33 @@ +import java.io.BufferedReader; +import java.io.IOException; +import java.io.InputStreamReader; +import java.util.Scanner; + +public class StorageDeviceLister { + + public static String device = ""; + public static String fullPath = ""; + + protected static String deviceList(Scanner UserInput) { + + try { + ProcessBuilder pb = new ProcessBuilder("lsblk"); + Process process = pb.start(); + + BufferedReader reader = new BufferedReader( + new InputStreamReader(process.getInputStream()) + ); + + String line; + while ((line = reader.readLine()) != null) { + System.out.println(line); + } + } catch (IOException e) { + e.printStackTrace(); + } + String device = UserInput.nextLine(); + System.out.println("Using device: " + "/dev/" + device); + fullPath = "/dev/" + device; + return fullPath; + } +} diff --git a/YesNo.java b/YesNo.java new file mode 100644 index 0000000..5280bc9 --- /dev/null +++ b/YesNo.java @@ -0,0 +1,16 @@ +public class YesNo { + public static boolean check(String input) { + switch (input) { + case "Y": + return true; + case "y": + return true; + case "": + return true; + case "n": + return false; + default: + return false; + } + } +}