Dépôt officiel du code source de l'ERP OpenConcerto

/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 2011 OpenConcerto, by ILM Informatique. All rights reserved.

 *

 * The contents of this file are subject to the terms of the GNU General Public License Version 3

 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a

 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific

 * language governing permissions and limitations under the License.

 *

 * When distributing the software, include this License Header Notice in each file.

 */

 package org.openconcerto.utils;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.io.UnsupportedEncodingException;

import java.net.InetAddress;

import java.net.NetworkInterface;

import java.net.ServerSocket;

import java.net.SocketException;

import java.net.URL;

import java.net.URLEncoder;

import java.nio.charset.StandardCharsets;

import java.security.KeyManagementException;

import java.security.NoSuchAlgorithmException;

import java.security.cert.X509Certificate;

import java.util.Enumeration;

import java.util.LinkedHashMap;

import java.util.Map;

import java.util.Map.Entry;

import java.util.regex.Pattern;

import javax.net.ssl.HostnameVerifier;

import javax.net.ssl.HttpsURLConnection;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLSession;

import javax.net.ssl.TrustManager;

import javax.net.ssl.X509TrustManager;

public class NetUtils {

    public static int findFreePort(final int preferred) {

        return findFreePort(null, preferred);

    }

    /**

     * Returns a free port number on localhost.

     *

     * @param addr the bind address, <code>null</code> meaning an address of the loopback interface.

     * @param preferred if this port is free, then it's returned.

     * @return preferred if free otherwise any free port.

     */

    public static int findFreePort(final String addr, final int preferred) {

        if (isPortFree(addr, preferred))

            return preferred;

        else

            return findFreePort(addr);

    }

    /**

     * Returns a free port number on localhost.

     *

     * @return a free port number on localhost, or -1 if unable to find a free port

     */

    public static int findFreePort() {

        return findFreePort(null);

    }

    public static int findFreePort(final String addr) {

        return checkPort(addr, 0);

    }

    public static boolean isPortFree(final String addr, final int port) {

        if (port <= 0)

            throw new IllegalArgumentException(port + " is negative");

        return checkPort(addr, port) == port;

    }

    // with code from from org/eclipse/jdt/launching/SocketUtil.java

    private static int checkPort(final String addr, final int port) {

        ServerSocket socket = null;

        try {

            // ATTN InetAddress.getByName(null) means an address of the loopback interface, while

            // passing null means any/all local addresses. The problem is that the constructor will

            // succeed for a given port even if it is already bound with the other address.

            socket = new ServerSocket(port, 0, InetAddress.getByName(addr));

            return socket.getLocalPort();

        } catch (IOException e) {

        } finally {

            if (socket != null) {

                try {

                    socket.close();

                } catch (IOException e) {

                }

            }

        }

        return -1;

    }

    /**

     * Whether the passed address refers to this computer.

     *

     * @param addr an ip or dns address, eg "192.168.28.52".

     * @return <code>true</code> if <code>addr</code> is bound to an interface of this computer.

     */

    static public final boolean isSelfAddr(String addr) {

        if (addr == null)

            return false;

        if (addr.startsWith("127.") || addr.startsWith("localhost"))

            return true;

        try {

            final Enumeration<NetworkInterface> en = NetworkInterface.getNetworkInterfaces();

            while (en.hasMoreElements()) {

                final NetworkInterface ni = en.nextElement();

                final Enumeration<InetAddress> addresses = ni.getInetAddresses();

                while (addresses.hasMoreElements()) {

                    final InetAddress inetAddress = addresses.nextElement();

                    if (addr.startsWith(inetAddress.getHostAddress()))

                        return true;

                }

            }

            return false;

        } catch (SocketException e) {

            e.printStackTrace();

            return false;

        }

    }

    public static final HostnameVerifier HostnameNonVerifier = new HostnameVerifier() {

        @Override

        public boolean verify(String hostname, SSLSession session) {

            return true;

        }

    };

    public static final String getHTTPContent(String address) {

        return getHTTPContent(address, false);

    }

    public static final String getHTTPContent(String address, final boolean dontVerify) {

        String content = "";

        OutputStream out = null;

        HttpsURLConnection conn = null;

        InputStream in = null;

        try {

            URL url = new URL(address);

            out = new ByteArrayOutputStream();

            conn = (HttpsURLConnection) url.openConnection();

            // Sur la connexion

            if (dontVerify) {

                conn.setHostnameVerifier(HostnameNonVerifier);

                // ou globalement

                // HttpsURLConnection.setDefaultHostnameVerifier(new CustomizedHostnameVerifier());

            }

            in = conn.getInputStream();

            final byte[] buffer = new byte[1024];

            int numRead;

            while ((numRead = in.read(buffer)) != -1) {

                out.write(buffer, 0, numRead);

            }

            content = out.toString();

        } catch (Exception exception) {

            exception.printStackTrace();

        } finally {

            try {

                if (in != null) {

                    in.close();

                }

                if (out != null) {

                    out.close();

                }

            } catch (IOException ioe) {

            }

        }

        return content;

    }

    static public final String urlEncode(final String... kv) {

        return urlEncode(false, kv);

    }

    static public final String urlEncode(final boolean spaceAsPlus, final String... kv) {

        final int size = kv.length;

        if (size % 2 != 0)

            throw new IllegalArgumentException("Odd number of items : " + size);

        final LinkedHashMap<String, Object> map = new LinkedHashMap<>(size / 2, 1);

        for (int i = 0; i < size; i += 2) {

            map.put(kv[i], kv[i + 1]);

        }

        return urlEncode(map, spaceAsPlus);

    }

    /**

     * Encode pairs in application/x-www-form-urlencoded format. Although both the

     * <a href="https://url.spec.whatwg.org/#urlencoded-serializing">URL standard</a> and the

     * <a href="https://www.w3.org/TR/html4/interact/forms.html#h-17.13.4.1">HTML standard</a>

     * specifies <code>spaceAsPlus</code> some applications expect the opposite (e.g. POST with

     * jetty).

     *

     * @param map the values to encode.

     * @param spaceAsPlus <code>true</code> to use '+' for a space, <code>false</code> to use '%20'.

     * @return the encoded string.

     * @see #percentEncode(String, String, boolean)

     */

    static public final String urlEncode(final Map<String, ?> map, final boolean spaceAsPlus) {

        if (map.isEmpty())

            return "";

        final String charset = StandardCharsets.UTF_8.name();

        final StringBuilder sb = new StringBuilder(256);

        for (final Entry<String, ?> e : map.entrySet()) {

            final Object value = e.getValue();

            // Avoid null and "null" confusion.

            if (value != null) {

                try {

                    sb.append(percentEncode(e.getKey(), charset, spaceAsPlus));

                    sb.append('=');

                    sb.append(percentEncode(String.valueOf(value), charset, spaceAsPlus));

                    sb.append('&');

                } catch (UnsupportedEncodingException exn) {

                    throw new IllegalStateException("UTF-8 should be standard", exn);

                }

            }

        }

        // remove last '&'

        sb.setLength(sb.length() - 1);

        return sb.toString();

    }

    static private final Pattern SPACE_PATTERN = Pattern.compile("+", Pattern.LITERAL);

    /**

     * Percent encode a string.

     *

     * @param s the string to encode.

     * @param charset the charset to use.

     * @param spaceAsPlus <code>true</code> to use '+' for a space (as per

     *        <a href="https://www.w3.org/TR/html4/interact/forms.html#h-17.13.4.1">HTML</a> and

     *        <a href=

     *        "https://url.spec.whatwg.org/#urlencoded-serializing">application/x-www-form-urlencoded</a>

     *        of URL Standard), <code>false</code> to use '%20' (as per

     *        <a href="https://url.spec.whatwg.org/#path-state">Path part</a> of URL Standard).

     * @return the encoded string.

     * @throws UnsupportedEncodingException if charset is unknown.

     * @see <a href="https://url.spec.whatwg.org/#string-percent-encode-after-encoding">URL

     *      standard</a>

     */

    static public final String percentEncode(final String s, final String charset, final boolean spaceAsPlus) throws UnsupportedEncodingException {

        final String withPlus = URLEncoder.encode(s, charset);

        if (spaceAsPlus)

            return withPlus;

        return SPACE_PATTERN.matcher(withPlus).replaceAll("%20");

    }

    // Create a trust manager that does not validate certificate chains

    static private final TrustManager[] TRUSTALL_MANAGERS = new TrustManager[] { new X509TrustManager() {

        public java.security.cert.X509Certificate[] getAcceptedIssuers() {

            return null;

        }

        public void checkClientTrusted(X509Certificate[] certs, String authType) {

        }

        public void checkServerTrusted(X509Certificate[] certs, String authType) {

        }

    } };

    static public final SSLContext createTrustAllContext() throws KeyManagementException, NoSuchAlgorithmException {

        final SSLContext sc = SSLContext.getInstance("TLS");

        sc.init(null, TRUSTALL_MANAGERS, new java.security.SecureRandom());

        return sc;

    }

}