-
Notifications
You must be signed in to change notification settings - Fork 198
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
285 additions
and
55 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,198 @@ | ||
| package burp; | ||
|
|
||
| import java.awt.Toolkit; | ||
| import java.awt.datatransfer.Clipboard; | ||
| import java.awt.datatransfer.StringSelection; | ||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.text.SimpleDateFormat; | ||
| import java.util.Arrays; | ||
| import java.util.Date; | ||
|
|
||
| import org.apache.commons.io.FileUtils; | ||
|
|
||
| /* | ||
| * 在系统terminal中执行命令，实现思路： | ||
| * 1、将命令写入bat文件 | ||
| * 2、通过执行bat文件执行命令 | ||
| */ | ||
| public class TerminalExec { | ||
|
|
||
| String workdir; | ||
| String cmdContent; | ||
| String FullBatchFilePath; | ||
|
|
||
| public String getWorkdir() { | ||
| return workdir; | ||
| } | ||
|
|
||
| public void setWorkdir(String workdir) { | ||
| this.workdir = workdir; | ||
| } | ||
|
|
||
| public String getCmdContent() { | ||
| return cmdContent; | ||
| } | ||
|
|
||
| public void setCmdContent(String cmdContent) { | ||
| this.cmdContent = cmdContent; | ||
| } | ||
|
|
||
| public String getFullBatchFilePath() { | ||
| return FullBatchFilePath; | ||
| } | ||
|
|
||
| public void setFullBatchFilePath(String fullBatchFilePath) { | ||
| FullBatchFilePath = fullBatchFilePath; | ||
| } | ||
|
|
||
| /* | ||
| * workdir --the dir of batch file | ||
| * | ||
| */ | ||
| public TerminalExec(String workdir, String batchFileName,String parserPath,String executerPath, String parameter){ | ||
| if (workdir == null) { | ||
| workdir = (String) System.getProperties().get("java.io.tmpdir"); | ||
| } | ||
| cmdContent = changeDirCommand(workdir); | ||
| cmdContent = cmdContent +genCmd(parserPath,executerPath,parameter); | ||
| FullBatchFilePath = genBatchFile(cmdContent,batchFileName); | ||
| } | ||
|
|
||
| public void run() { | ||
| executeBatchFile(FullBatchFilePath); | ||
| } | ||
|
|
||
| /* | ||
| * 通知执行bat文件来执行命令 | ||
| */ | ||
| public static Process executeBatchFile(String batfilepath) { | ||
| String command = ""; | ||
| if (Utils.isWindows()) { | ||
| command="cmd /c start " + batfilepath; | ||
| } else { | ||
| if (new File("/bin/sh").exists()) { | ||
| command="/bin/sh " + batfilepath; | ||
| } | ||
| else if (new File("/bin/bash").exists()) { | ||
| command="/bin/bash " + batfilepath; | ||
| } | ||
| } | ||
| try { | ||
| Process process = Runtime.getRuntime().exec(command); | ||
| return process; | ||
| } catch (IOException e) { | ||
| e.printStackTrace(); | ||
| return null; | ||
| } | ||
| } | ||
|
|
||
| public String genBatchFile(String cmdContent, String batchFileName) { | ||
| try { | ||
| //将命令写入剪切板 | ||
| Clipboard clipboard = Toolkit.getDefaultToolkit().getSystemClipboard(); | ||
| StringSelection selection = new StringSelection(cmdContent); | ||
| clipboard.setContents(selection, null); | ||
|
|
||
| if (batchFileName == null || batchFileName.trim().equals("")) { | ||
| SimpleDateFormat simpleDateFormat = new SimpleDateFormat("MMdd-HHmmss"); | ||
| String timeString = simpleDateFormat.format(new Date()); | ||
| batchFileName = timeString+".bat"; | ||
| }else if(!batchFileName.endsWith(".bat") && !batchFileName.endsWith(".cmd")) { | ||
| batchFileName = batchFileName+".bat"; | ||
| } | ||
|
|
||
| File batFile = new File(workdir,batchFileName); | ||
| if (!batFile.exists()) { | ||
| batFile.createNewFile(); | ||
| } | ||
|
|
||
| FileUtils.writeByteArrayToFile(batFile, cmdContent.getBytes()); | ||
| return batFile.getAbsolutePath(); | ||
| } catch (IOException e) { | ||
| e.printStackTrace(); | ||
| return null; | ||
| } | ||
| } | ||
|
|
||
| /* | ||
| * 切换工作目录 | ||
| */ | ||
| public String changeDirCommand(String dir){ | ||
| //运行命令的工作目录，work path | ||
| String command = "cd "+dir+System.lineSeparator(); | ||
|
|
||
| if (Utils.isWindows()) {//如果是windows，还要注意不同磁盘的切换 | ||
| String diskString = dir.split(":")[0]; | ||
| command =command+ diskString+":"+System.lineSeparator(); | ||
| } | ||
| return command; | ||
| } | ||
|
|
||
| /* | ||
| * parserPath --- python.exe java.exe .... | ||
| * executerPath --- sqlmap.py nmap.exe .... | ||
| * parameters ---- -v -A -r xxx.file ..... | ||
| */ | ||
| public static String genCmd(String parserPath,String executerPath, String parameter) { | ||
| StringBuilder command = new StringBuilder(); | ||
|
|
||
| if ((parserPath != null && new File(parserPath).exists() && new File(parserPath).isFile()) | ||
| || isInEnvironmentPath(parserPath)){ | ||
|
|
||
| if (parserPath.contains(" ")) { | ||
| parserPath = "\""+parserPath+"\"";//如果路径中包含空格，需要引号 | ||
| } | ||
| command.append(parserPath); | ||
| command.append(" "); | ||
| } | ||
|
|
||
| if ((executerPath != null && new File(executerPath).exists() && new File(executerPath).isFile()) | ||
| || isInEnvironmentPath(executerPath)){ | ||
|
|
||
| if (executerPath.contains(" ")) { | ||
| executerPath = "\""+executerPath+"\"";//如果路径中包含空格，需要引号 | ||
| } | ||
|
|
||
| command.append(executerPath); | ||
| command.append(" "); | ||
| } | ||
|
|
||
| if (parameter != null && !parameter.equals("")) { | ||
| command.append(parameter); | ||
| } | ||
| command.append(System.lineSeparator()); | ||
| return command.toString(); | ||
| } | ||
|
|
||
| /* | ||
| * 判断某个文件是否在环境变量中 | ||
| */ | ||
| public static boolean isInEnvironmentPath(String filename) { | ||
| if (filename == null) { | ||
| return false; | ||
| } | ||
| String pathvalue = System.getenv().get("Path"); | ||
| // System.out.println(pathvalue); | ||
| String[] items = pathvalue.split(";"); | ||
| for (String item:items) { | ||
| File tmpPath = new File(item); | ||
| if (tmpPath.isDirectory()) { | ||
| // System.out.println(Arrays.asList(tmpPath.listFiles())); | ||
| File fullpath = new File(item,filename); | ||
| if (Arrays.asList(tmpPath.listFiles()).contains(fullpath)) { | ||
| return true; | ||
| }else { | ||
| continue; | ||
| } | ||
| } | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| public static void main(String[] args) { | ||
| System.out.println(isInEnvironmentPath("nmap.exe")); | ||
| TerminalExec xxx = new TerminalExec(null,"nmap-test.bat",null,"nmap.exe","-v -A www.baidu.com"); | ||
| xxx.run(); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.