psiprobe.tools.Whois 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.tools;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.TreeMap;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import psiprobe.tools.url.UrlParser;

/**
 * The Class Whois.
 */
public final class Whois {

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

  /**
   * Prevent Instantiation.
   */
  private Whois() {
    // Prevent Instantiation
  }

  /**
   * Lookup.
   *
   * @param server the server
   * @param port the port
   * @param query the query
   *
   * @return the response
   *
   * @throws IOException Signals that an I/O exception has occurred.
   */
  public static Response lookup(String server, int port, String query) throws IOException {
    return lookup(server, port, query, 5);
  }

  /**
   * Lookup.
   *
   * @param server the server
   * @param port the port
   * @param query the query
   * @param timeout the timeout
   *
   * @return the response
   *
   * @throws IOException Signals that an I/O exception has occurred.
   */
  public static Response lookup(String server, int port, String query, long timeout)
      throws IOException {

    return lookup(server, port, query, timeout, System.lineSeparator());
  }

  /**
   * Lookup.
   *
   * @param server the server
   * @param port the port
   * @param query the query
   * @param timeout the timeout
   * @param lineSeparator the line separator
   *
   * @return the response
   *
   * @throws IOException Signals that an I/O exception has occurred.
   */
  public static Response lookup(String server, int port, String query, long timeout,
      String lineSeparator) throws IOException {

    if (query == null) {
      return null;
    }

    Response response = new Response();

    response.server = server;
    response.port = port;

    try (Socket connection = AsyncSocketFactory.createSocket(server, port, timeout);
        PrintStream out =
            new PrintStream(connection.getOutputStream(), true, StandardCharsets.UTF_8.name());
        BufferedReader in = new BufferedReader(
            new InputStreamReader(connection.getInputStream(), StandardCharsets.UTF_8))) {
      out.println(query);
      StringBuilder sb = new StringBuilder();

      String line;
      while ((line = in.readLine()) != null) {
        sb.append(line).append(lineSeparator);
        line = line.trim();
        if (!line.startsWith("%") && !line.startsWith("#")) {
          int fs = line.indexOf(':');
          if (fs > 0) {
            String name = line.substring(0, fs);
            String value = line.substring(fs + 1).trim();
            response.data.put(name, value);
          }
        }
      }
      response.summary = sb.toString();

      Response newResponse = null;
      String referral = response.getData().get("ReferralServer");

      if (referral != null) {
        try {
          UrlParser url = new UrlParser(referral);
          if ("whois".equals(url.getProtocol())) {
            newResponse = lookup(url.getHost(), url.getPort() == -1 ? 43 : url.getPort(), query,
                timeout, lineSeparator);
          }
        } catch (IOException e) {
          logger.trace("Could not contact '{}'", referral, e);
        }
      }
      if (newResponse != null) {
        response = newResponse;
      }
    }

    return response;
  }

  /**
   * The Class Response.
   */
  public static final class Response {

    /** The summary. */
    String summary;

    /** The data. */
    Map data = new TreeMap<>();

    /** The server. */
    String server;

    /** The port. */
    int port;

    /**
     * Gets the summary.
     *
     * @return the summary
     */
    public String getSummary() {
      return summary;
    }

    /**
     * Gets the data.
     *
     * @return the data
     */
    public Map getData() {
      return data;
    }

    /**
     * Gets the server.
     *
     * @return the server
     */
    public String getServer() {
      return server;
    }

    /**
     * Gets the port.
     *
     * @return the port
     */
    public int getPort() {
      return port;
    }

    /**
     * Instantiates a new response.
     */
    private Response() {}

  }

}