Added config, clean code, and optimized imports.

RealSuperUser · RealSuperUser · commit bf5e23d5c5e7 · 2024-03-14T23:48:40.000+03:30
diff --git a/src/main/java/ir/xenoncommunity/jss/JSSAttack.java b/src/main/java/ir/xenoncommunity/jss/JSSAttack.java
@@ -7,6 +7,7 @@
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
 import lombok.SneakyThrows;
+import lombok.val;
 import org.jetbrains.annotations.NotNull;
 
 import java.net.InetAddress;
@@ -44,6 +45,7 @@ private IAttackMethod getMethod(final String method, final Integer byteSize, Att
      */
     @SneakyThrows
     public void run() {
+        // Create local variables for our configuration/run args
         boolean debug = false;
         String ip = null;
         Integer port;
@@ -53,19 +55,24 @@ public void run() {
         Integer duration;
         String method = null;
         Boolean verbose = null;
-        FileManager.values.add(new Value("ip", "0.0.0.0"));
-        FileManager.values.add(new Value("port", 1024));
-        FileManager.values.add(new Value("ppsLimit", 1000));
-        FileManager.values.add(new Value("maxThreads", 5));
-        FileManager.values.add(new Value("byteSize", 1024));
-        FileManager.values.add(new Value("method", "TCPFLOOD"));
-        FileManager.values.add(new Value("verbose", false));
-        FileManager.values.add(new Value("duration", -1));
+        // Check if user is using config
         if (this.parser.get("--config", String.class, null) != null){
+            // to solve the error, there is no purpose
             duration = null;
             port = null;
+            // Add values to FileManager
+            FileManager.values.add(new Value<>("ip", "0.0.0.0"));
+            FileManager.values.add(new Value<>("port", 1024));
+            FileManager.values.add(new Value<>("ppsLimit", 1000));
+            FileManager.values.add(new Value<>("maxThreads", 5));
+            FileManager.values.add(new Value<>("byteSize", 1024));
+            FileManager.values.add(new Value<>("method", "TCPFLOOD"));
+            FileManager.values.add(new Value<>("verbose", false));
+            FileManager.values.add(new Value<>("duration", -1));
+            // Init the FileManager
             FileManager.init();
-            for(Value val : FileManager.values){
+            // Switch between configuration items and set the values
+            for(val val : FileManager.values){
                 switch(val.getName()){
                     case "ip":{
                         ip = (String) val.getValue();
@@ -102,9 +109,9 @@ public void run() {
 
                 }
             }
-
+        // See If user isn't using configuration and using run args instead
         } else {
-            //Clears the entire FileManager values because we don't need them.
+            // Clears the entire FileManager values because we don't need them
             FileManager.values.clear();
             // Parse command line arguments
             debug = this.parser.get("--debug", Boolean.class, false);
@@ -118,16 +125,17 @@ public void run() {
             verbose = this.parser.get("--verbose", Boolean.class, false);
 
         }
+        // Check if values are null
         if (ip == null) {
             System.out.println("JSSAttack by XenonCommunity");
             System.out.println("Usage: java -jar JSSAttack.jar --ip <ip> --port <port> --threads <threads> --byteSize <byteSize> --duration <duration> --method <method> [--verbose] [--debug]");
             System.exit(0);
         }
 
         // Set logging level based on verbosity and debug mode
-        if (verbose) {
+        if (verbose != null && verbose) {
             Logger.setCurrentLevel(Logger.LEVEL.VERBOSE);
-        } else if (debug) {
+        } else if (debug){
             Logger.setCurrentLevel(Logger.LEVEL.DEBUG);
         }
 
@@ -139,12 +147,13 @@ public void run() {
         Logger.log(Logger.LEVEL.DEBUG, "ByteSize is: " + byteSize);
         Logger.log(Logger.LEVEL.DEBUG, "Duration is: " + duration);
         Logger.log(Logger.LEVEL.DEBUG, "Method is: " + method);
-
+        assert maxThreads != null;
         // Initialize task manager
         TaskManager taskManager = new TaskManager(maxThreads + 1);
 
         // Initialize attack parameters
         final AttackStatics attackStatics = new AttackStatics(ppsLimit);
+        assert method != null;
         final IAttackMethod attackMethod = getMethod(method, byteSize, attackStatics);
         final InetAddress addr = InetAddress.getByName(ip);
         Integer finalDuration = duration;
diff --git a/src/main/java/ir/xenoncommunity/jss/Main.java b/src/main/java/ir/xenoncommunity/jss/Main.java
@@ -1,8 +1,6 @@
 package ir.xenoncommunity.jss;
 
 import ir.xenoncommunity.jss.utils.CommandLineParser;
-import ir.xenoncommunity.jss.utils.FileManager;
-import ir.xenoncommunity.jss.utils.filemanager.Value;
 
 public class Main {
     /**
diff --git a/src/main/java/ir/xenoncommunity/jss/methods/impl/HTTPFlood.java b/src/main/java/ir/xenoncommunity/jss/methods/impl/HTTPFlood.java
@@ -9,7 +9,6 @@
 import lombok.val;
 import org.jetbrains.annotations.NotNull;
 
-import java.io.OutputStream;
 import java.net.InetAddress;
 import java.net.Socket;
 import java.net.StandardSocketOptions;
diff --git a/src/main/java/ir/xenoncommunity/jss/methods/impl/TCPFlood.java b/src/main/java/ir/xenoncommunity/jss/methods/impl/TCPFlood.java
@@ -8,7 +8,6 @@
 import lombok.RequiredArgsConstructor;
 import lombok.val;
 
-import java.io.OutputStream;
 import java.net.InetAddress;
 import java.net.Socket;
 import java.net.StandardSocketOptions;
diff --git a/src/main/java/ir/xenoncommunity/jss/utils/CommandLineParser.java b/src/main/java/ir/xenoncommunity/jss/utils/CommandLineParser.java
@@ -1,15 +1,12 @@
 package ir.xenoncommunity.jss.utils;
 
+import lombok.AllArgsConstructor;
 import org.jetbrains.annotations.Nullable;
 
+@AllArgsConstructor
 public class CommandLineParser {
 
     private final String[] args;
-
-    public CommandLineParser(final String[] args) {
-        this.args = args;
-    }
-
     public <T> T get(final String arg, final Class<T> type, final @Nullable T defaultValue) {
         for (int i = 0; i < args.length - 1; i++) {
             if (!args[i].equals(arg)) continue;
@@ -22,14 +19,15 @@ public <T> T get(final String arg, final Class<T> type, final @Nullable T defaul
 
 
     private <T> T convertToType(final String value, final Class<T> type) {
-        if (type.equals(String.class)) {
-            return type.cast(value);
-        } else if (type.equals(Integer.class)) {
-            return type.cast(Integer.parseInt(value));
-        } else if (type.equals(Boolean.class)) {
-            return type.cast(Boolean.parseBoolean(value));
-        } else {
-            throw new IllegalArgumentException("Unsupported type");
-        }
+        return type.equals(String.class) ?
+                type.cast(value)
+                :
+                (type.equals(Integer.class)
+                ?
+                type.cast(Integer.parseInt(value))
+                :
+                (type.equals(Boolean.class)
+                ? type.cast(Boolean.parseBoolean(value))
+                : null));
     }
 }
diff --git a/src/main/java/ir/xenoncommunity/jss/utils/FileManager.java b/src/main/java/ir/xenoncommunity/jss/utils/FileManager.java
@@ -1,18 +1,23 @@
 package ir.xenoncommunity.jss.utils;
 
 import ir.xenoncommunity.jss.utils.filemanager.Value;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.SneakyThrows;
 import lombok.experimental.UtilityClass;
-import lombok.*;
+import lombok.val;
 
 import java.io.*;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
 @Getter
 @Setter
 @UtilityClass
 public class FileManager {
     public final File folder = new File("JSS");
     public final File config = new File(folder,"config.txt");
     public final List<Value> values = new ArrayList<>();
+    private boolean isSuccess = true;
     public void init(){
         if(folder.exists() && config.exists()){
             readConf();
@@ -32,8 +37,8 @@ public void readConf(){
     }
     @SneakyThrows
     public void saveForFirstTime(){
-        folder.mkdirs();
-        config.createNewFile();
+        isSuccess = folder.mkdirs();
+        isSuccess = config.createNewFile();
         val bw = new BufferedWriter(new FileWriter(config));
         values.forEach(e -> {
             try {
@@ -42,5 +47,12 @@ public void saveForFirstTime(){
             } catch (Exception ignored) {}
         });
         bw.close();
+        isFaild();
+    }
+    private void isFaild(){
+        Logger.log(Logger.LEVEL.ERROR, "Please rerun the program.\n" +
+                "this caused due to lack of permissions,\n" +
+                "so the program wasn't able to create config files.");
+        System.exit(0);
     }
 }
diff --git a/src/main/java/ir/xenoncommunity/jss/utils/filemanager/Value.java b/src/main/java/ir/xenoncommunity/jss/utils/filemanager/Value.java
@@ -4,8 +4,6 @@
 import lombok.Getter;
 import lombok.Setter;
 
-import java.util.Objects;
-
 @Getter
 @Setter
 @AllArgsConstructor
