一、如何运行python程序
java中运行python程序大致有两种思路,一种是利用jython的环境进行java和python交互,第二种是利用java内的Process类模拟系统终端的操作,我们这里利用Process类进行交互。
大体操作如下:
int exitValue;
try {
Process pro = Runtime.getRuntime().exec("python", "python文件名", "其他参数");
InputStream out = pro.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(in));
String line;
while ((line = bufferedReader.readLine()) != null) {
System.out.println(line);
}
pro.waitFor();
exitValue = pro.exitValue();
} catch (IOException | InterruptedException e) {
exitValue = 2;
}
System.out.println("exit: " + exitValue);
但是,如果遇到像sqlmap这样需要中途用户进行选择的话,就没有办法进行输入操作。虽然Process类中有getOutputStream的方法,但是如果没有到需要输入的地方输入无效,而且这个线程会在
bufferedReader.readLine()中挂起(终端中等待输入没有"\n",意味着readLine没有结束),后续操作无法执行。
二、解决方法(仅对sqlmap有效)
根据已知问题,由sqlmap对用户选择的语法分析发现会出现类似“
[Y/N] ”的字段,这个字段之后就是用户输入的地方,我们利用这一点对代码进行改造:
int exitValue; try {
Process pro = Runtime.getRuntime().exec("python", "python文件名", "其他参数");InputStream in = pro.getInputStream();OutputStream out = pro.getOutputStream();BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(in));int charNum; StringBuilder sb = new StringBuilder(); while ((charNum = bufferedReader.read()) != -1) { sb.append((char) charNum); System.out.print((char) charNum); if (charNum == '\n' || charNum == '\r') { sb = new StringBuilder(); } if (sb.toString().endsWith("[Y/n] ") || sb.toString().endsWith("[Y/N] ") || sb.toString().endsWith("[y/n] ") || sb.toString().endsWith("[y/N] ")) { Scanner scanner = new Scanner(System.in); String input = scanner.next(); out.write((input + "\n").getBytes(Charset.defaultCharset())); out.flush(); } } pro.waitFor(); exitValue = pro.exitValue(); } catch (IOException | InterruptedException e) { exitValue = 2; }
这段代码则是逐字符读入readLine方法防止因没有换行符挂起的问题,其次,根据sqlmap的语法,在询问用户是会出现上述类似字段,则判断现有字段中是否已经存在上述字段,如果存在说明将会等待用户输入,这是利用OutputStream输出内容,其中“\n”将会指示“终端”输入完成并进行后续操作。
这样则可以基本解决sqlmap的输入问题。
第一篇博客留影纪念。