SpringBoot + nmap4j 获取端口信息

publicclassNmap4jTest{
@Test
publicvoidbasicNmap4jUsageTest(){

try {
               // 这里的路径要改为刚才 nmap 软件的安装路径
   String path = "/usr/local" ;
   
   Nmap4j nmap4j = new Nmap4j( path ) ;
               // 这地方使用了 -oX 后面要跟文件名称
   nmap4j.addFlags( "-sV -T5 -O -oX -" ) ;
   nmap4j.includeHosts( "localhost" ) ;
   nmap4j.execute() ;
   if( !nmap4j.hasError() ) {
                        
    NMapRun nmapRun = nmap4j.getResult() ;  // 这一行一定要注释掉，不然会一直报错
    String output = nmap4j.getOutput() ;  // 这一样代码意义也不大，我直接删掉了
    if( output == null ) {
     fail() ;
    }
    String errors = nmap4j.getExecutionResults().getErrors() ;
    if (errors == null ) {
     fail() ;
    }
   }
  } catch (NMapInitializationException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
   fail() ;
  } catch (NMapExecutionException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
   fail() ;
  }

 }

}

/**
 * 使用 nmap4j 工具进行扫描
 *
 * @param ip    目标 ip
 * @param ports 目标端口
 * @return 端口信息列表
 */
@RequestMapping("/querydb")
public List<NmapPortInfo> querydb(@RequestParam(value = "ip") String ip, @RequestParam("ports") List<String> ports) {
    ArrayList<NmapPortInfo> portInfos = new ArrayList<>();
    // 1.拼接端口
    String portStr = StrUtil.join(",", ports);
    //2. 指定 nmap 路径
    String path = "D:/StudyApps/nmap";
    String fileName = "temp_result.xml";

    Nmap4j nmap4j = new Nmap4j(path);

    //3.读取端口耗时较长，可以使用异步
    CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
        nmap4j.addFlags("-sV -p " + portStr + " -T5 -O -oX " + fileName);
        nmap4j.includeHosts(ip);
        try {
            nmap4j.execute();
        } catch (Exception e) {
            thrownew RuntimeException(e);
        }
    }, threadPoolExecutor);

    future.join();

    //4. 获取端口信息
    return getPortInfo(portInfos, fileName);
}
/**
     * 获取 ip + 端口信息,封装为集合返回前端
     *
     * @param portInfos 返回前端集合
     * @param fileName  临时的 xml 文件
     * @return 信息列表
     */
    @SneakyThrows
    private List<NmapPortInfo> getPortInfo(List<NmapPortInfo> portInfos, String fileName){
        // 获取项目所在路径
        String projectPath = System.getProperty("user.dir");
        // 拼接文件路径
        String filePath = projectPath + FileUtil.FILE_SEPARATOR + fileName;
        log.info("文件路径：{}", filePath);

        // nmap 返回 xml 格式固定，使用 dom4j 解析
        SAXReader reader = new SAXReader();
        org.dom4j.Document document = reader.read(FileUtil.file(filePath));
        org.dom4j.Element rootElement = document.getRootElement();
        org.dom4j.Element element = rootElement.element("host");

        org.dom4j.Element xmlPorts = element.element("ports");

        List<org.dom4j.Element> port = xmlPorts.elements("port");
        for (org.dom4j.Element port1 : port) {
            Element service = port1.element("service");
            String product = service.attributeValue("product");
            String version = service.attributeValue("version");
            NmapPortInfo nmapPortInfo = new NmapPortInfo(product, version);
            portInfos.add(nmapPortInfo);
        }

        // 删除临时文件
        FileUtil.del(filePath);
        return portInfos;
    }
}

/**
 * 使用 nmap4j 工具进行扫描, linux系统
 *
 * @param ip    目标 ip
 * @param ports 目标端口
 * @return 端口信息列表
 */
@GetMapping("/linux/querydb")
@SneakyThrows
public List<NmapPortInfo> linuxQuerydb(@RequestParam(value = "ip") String ip, @RequestParam("ports") List<String> ports) {
    ArrayList<NmapPortInfo> portInfos = new ArrayList<>();
    // 1.拼接端口
    String portStr = StrUtil.join(",", ports);
    String fileName = "temp_result.xml";
    //2. linux namp 命令
    String nmapCommand = "nmap -sV -p " + portStr + " -T5 -O -oX " + fileName + " " + ip;

    //3. 读取端口耗时较长，可以使用异步
    CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
        Process nampProcess = null;
        try {
            // 3. 运行命令
            nampProcess = Runtime.getRuntime().exec(nmapCommand);
            // 4. 等待命令执行完成
            nampProcess.waitFor();
        } catch (Exception e) {
            thrownew RuntimeException(e);
        }
    }, threadPoolExecutor);

    future.join();

    // 5. 获取端口信息
    return getPortInfo(portInfos, fileName);
}