psiprobe.controllers.WhoisController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of psi-probe-core Show documentation
PSI Probe Core - Core logic, data models, and controllers
There is a newer version: 5.1.1
/**
 * Licensed under the GPL License. You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE.
 */
package psiprobe.controllers;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.ServletRequestUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.ParameterizableViewController;
import psiprobe.tools.Whois;

/**
 * The Class WhoisController.
 */
@Controller
public class WhoisController extends ParameterizableViewController {

  /** The Constant logger. */
  private static final Logger logger = LoggerFactory.getLogger(WhoisController.class);

  /** The lookup timeout. */
  private long lookupTimeout;

  /** The default server. */
  private String defaultServer;

  /** The default port. */
  private int defaultPort;

  /**
   * Gets the lookup timeout.
   *
   * @return the lookup timeout
   */
  public long getLookupTimeout() {
    return lookupTimeout;
  }

  /**
   * Sets the lookup timeout.
   *
   * @param lookupTimeout the new lookup timeout in seconds
   */
  @Value("5")
  public void setLookupTimeout(long lookupTimeout) {
    this.lookupTimeout = lookupTimeout;
  }

  /**
   * Gets the default server.
   *
   * @return the default server
   */
  public String getDefaultServer() {
    return defaultServer;
  }

  /**
   * Sets the default server.
   *
   * @param defaultServer the new default server
   */
  @Value("whois.arin.net")
  public void setDefaultServer(String defaultServer) {
    this.defaultServer = defaultServer;
  }

  /**
   * Gets the default port.
   *
   * @return the default port
   */
  public int getDefaultPort() {
    return defaultPort;
  }

  /**
   * Sets the default port.
   *
   * @param defaultPort the new default port
   */
  @Value("43")
  public void setDefaultPort(int defaultPort) {
    this.defaultPort = defaultPort;
  }

  @RequestMapping(path = "/whois.ajax")
  @Override
  public ModelAndView handleRequest(HttpServletRequest request, HttpServletResponse response)
      throws Exception {
    return super.handleRequest(request, response);
  }

  @Override
  protected ModelAndView handleRequestInternal(HttpServletRequest request,
      HttpServletResponse response) throws Exception {

    boolean timeout = false;

    String ipAddress = ServletRequestUtils.getStringParameter(request, "ip", null);

    Whois.Response wh = null;
    try {
      wh = Whois.lookup(getDefaultServer(), getDefaultPort(), ipAddress, getLookupTimeout());
    } catch (IOException e) {
      timeout = true;
      logger.trace("", e);
    }

    List lines = null;
    if (wh != null) {
      lines = new ArrayList<>(50);
      try (BufferedReader br = new BufferedReader(new InputStreamReader(
          new ByteArrayInputStream(wh.getSummary().getBytes(StandardCharsets.UTF_8)),
          StandardCharsets.UTF_8))) {
        String line;
        while ((line = br.readLine()) != null) {
          lines.add(line);
        }
      }
    }

    String reverseName = null;
    if (ipAddress != null) {
      try {
        reverseName = InetAddress.getByName(ipAddress).getCanonicalHostName();
      } catch (UnknownHostException e) {
        logger.error("could not run a DNS query on {}", ipAddress);
        logger.trace("", e);
      }
    }
    return new ModelAndView(getViewName(), "result", lines).addObject("timeout", timeout)
        .addObject("whoisServer",
            wh != null ? wh.getServer() + ":" + wh.getPort() : defaultServer + ":" + defaultPort)
        .addObject("domainName", reverseName);
  }

  @Value("ajax/whois")
  @Override
  public void setViewName(String viewName) {
    super.setViewName(viewName);
  }

}