Method RunCmd

src/main/java/com/sumsec/core/ast/util/DotHandler.java:31–92

(String[] dotPath)

public class DotHandler  {
    private static Logger logger = LogManager.getLogger(DotHandler.class);
    public String[] RunCmd(String[] dotPath){
        boolean isWindowsOS = isWindowsOS();
        String[] cmd;
        mkdir(ConstatField.ResultTemp);
        ArrayList<String> imgPaths = new ArrayList<>();

        for (int i = 0; i < dotPath.length; i++) {
            String imgPath = ConstatField.ASTResultTemp + ConstatField.separator + dotPath[i].substring(dotPath[i].lastIndexOf("\\") + 1, dotPath[i].lastIndexOf(".")) + ".png";

            if (isWindowsOS) {
                //windows下的命令
                cmd = new String[]{"cmd.exe", "/c", "dot", "-Tpng", dotPath[i], "-o", imgPath};
            } else {
                //linux下的命令

                // fix #issue2
                // on unix use / instead \
                // has tow slash in imgPath
                imgPath = ConstatField.ASTResultTemp + ConstatField.separator + dotPath[i].substring(dotPath[i].lastIndexOf("/") + 1, dotPath[i].lastIndexOf(".")) + ".png";
                // fixme: use `/usr/bash -c` may has problem in some bash such as zsh
                cmd = new String[]{"dot", "-Tpng" , dotPath[i] , "-o", imgPath};
            }
            try {
                logger.info("开始生成图片 " + dotPath[i]);
                logger.info("正在执行命令 "+ Arrays.toString(cmd));
                sleep(1000);
                String charsetName = isWindowsOS ? "GBK" : "UTF-8";
                Runtime.getRuntime().exec(cmd);

                byte[] bytes = null;
                Scanner runCmd = new Scanner(Runtime.getRuntime().exec(cmd).getInputStream(), charsetName).useDelimiter("\\A");
                //fix issue#2 : on linux ,detect hasNext otherwise will throw an NoSuchElementException
                if (runCmd.hasNext()) {
                    bytes = runCmd.next().getBytes(charsetName);
                }


                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);
                    alert.setTitle("警告");
                    alert.setContentText(new String(bytes, charsetName));
                    alert.show();
                } else{
                    logger.info("生成图片成功");
                    logger.info("图片路径为：" + imgPath);
                    imgPaths.add(imgPath);
                }
            } catch (Exception e) {
                logger.error("生成图片失败 " + dotPath[i]);
                Alert alert = new Alert(Alert.AlertType.ERROR);
                alert.setTitle("Error");
                alert.setContentText(e.getMessage());
                alert.show();
                logger.error(e.getMessage());
            }