java中运行sqlmap中等待输入的问题

一、如何运行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的输入问题。

第一篇博客留影纪念。