Parser

SummerSec · Mar 1, 2022 · b8dd9f5 · b8dd9f5
1 parent 72459ee
commit b8dd9f5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/main/java/com/sumsec/core/ast/util/DotHandler.java b/src/main/java/com/sumsec/core/ast/util/DotHandler.java
@@ -41,7 +41,7 @@ public String[] RunCmd(String[] dotPath){
                 cmd = new String[]{"cmd.exe", "/c", "dot", "-Tpng", dotPath[i], "-o", imgPath};
             } else {
                 //linux下的命令
-                cmd = new String[]{"/bin/sh", "-c", "dot", "-Tpng" + dotPath[i] + "-o", imgPath};
+                cmd = new String[]{"/bin/sh", "-c", "dot", "-Tpng" , dotPath[i] , "-o", imgPath};
             }
             try {
 
@@ -50,7 +50,8 @@ public String[] RunCmd(String[] dotPath){
                 sleep(1000);
                 String charsetName = isWindowsOS ? "GBK" : "UTF-8";
                 byte[] bytes = new Scanner(Runtime.getRuntime().exec(cmd).getInputStream(), charsetName).useDelimiter("\\A").next().getBytes(charsetName);
-                if (bytes.length > 17) {
+                File file = new File(imgPath);
+                if (!file.exists()) {
                     logger.info("bytes.length = " + bytes.length);
                     logger.info("something wrong");
                     Alert alert = new Alert(Alert.AlertType.WARNING);

diff --git a/src/main/java/com/sumsec/core/cfg/uitls/OSUtil.java b/src/main/java/com/sumsec/core/cfg/uitls/OSUtil.java
@@ -5,6 +5,7 @@
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
+import java.io.File;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Scanner;
@@ -47,7 +48,7 @@ public String[] RunCmd(String[] dotPath){
                 cmd = new String[]{"cmd.exe", "/c", "dot", "-Tpng", dotPath[i], "-o", imgPath};
             } else {
                 //linux下的命令
-                cmd = new String[]{"/bin/sh", "-c", "dot", "-Tpng" + dotPath[i] + "-o", imgPath};
+                cmd = new String[]{"/bin/sh", "-c", "dot", "-Tpng" , dotPath[i] ,"-o", imgPath};
             }
             try {
 
@@ -56,7 +57,8 @@ public String[] RunCmd(String[] dotPath){
                 sleep(1000);
                 String charsetName = isWindowsOS ? "GBK" : "UTF-8";
                 byte[] bytes = new Scanner(Runtime.getRuntime().exec(cmd).getInputStream(), charsetName).useDelimiter("\\A").next().getBytes(charsetName);
-                if ( bytes.length > 17) {
+                File file = new File(imgPath);
+                if (!file.exists()) {
                     logger.info("bytes.length = " + bytes.length);
                     logger.info("something wrong");
                     Alert alert = new Alert(Alert.AlertType.WARNING);