GitNex/app/src/main/java/org/mian/gitnex/helpers/MemorizingTrustManager.java

package org.mian.gitnex.helpers;

import android.annotation.TargetApi;
import android.app.Activity;
import android.app.Notification;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.content.Context;
import android.content.Intent;
import android.content.SharedPreferences;
import android.net.Uri;
import android.util.Base64;
import android.util.SparseArray;
import android.os.Build;
import android.os.Handler;

import androidx.core.app.NotificationCompat;
import org.mian.gitnex.R;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.security.cert.*;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Enumeration;
import java.util.List;
import java.util.Locale;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLSession;
import javax.net.ssl.TrustManager;
import javax.net.ssl.TrustManagerFactory;
import javax.net.ssl.X509TrustManager;

/**
 * Author Georg Lukas, modified by anonTree1417
 */

public class MemorizingTrustManager implements X509TrustManager {

	private final static String DECISION_INTENT = "de.duenndns.ssl.DECISION";
	final static String DECISION_INTENT_ID = DECISION_INTENT + ".decisionId";
	final static String DECISION_INTENT_CERT = DECISION_INTENT + ".cert";

	private final static Logger LOGGER = Logger.getLogger(MemorizingTrustManager.class.getName());
	final static String DECISION_TITLE_ID = DECISION_INTENT + ".titleId";
	private final static int NOTIFICATION_ID = 100509;

	private final static String KEYSTORE_NAME = "keystore";
	private final static String KEYSTORE_KEY = "keystore";

	private Context context;
	private Activity foregroundAct;
	private NotificationManager notificationManager;
	private static int decisionId = 0;
	private static final SparseArray<MTMDecision> openDecisions = new SparseArray<>();

	private Handler masterHandler;
	private SharedPreferences keyStoreStorage;
	private KeyStore appKeyStore;
	private X509TrustManager defaultTrustManager;
	private X509TrustManager appTrustManager;

	/**
	 * Creates an instance of the MemorizingTrustManager class that falls back to a custom TrustManager.
	 * <p>
	 * You need to supply the application context. This has to be one of:
	 * - Application
	 * - Activity
	 * - Service
	 * <p>
	 * The context is used for file management, to display the dialog /
	 * notification and for obtaining translated strings.
	 *
	 * @param m                   Context for the application.
	 * @param defaultTrustManager Delegate trust management to this TM. If null, the user must accept every certificate.
	 */
	public MemorizingTrustManager(Context m, X509TrustManager defaultTrustManager) {

		init(m);
		this.appTrustManager = getTrustManager(appKeyStore);
		this.defaultTrustManager = defaultTrustManager;
	}

	/**
	 * Creates an instance of the MemorizingTrustManager class using the system X509TrustManager.
	 * <p>
	 * You need to supply the application context. This has to be one of:
	 * - Application
	 * - Activity
	 * - Service
	 * <p>
	 * The context is used for file management, to display the dialog /
	 * notification and for obtaining translated strings.
	 *
	 * @param m Context for the application.
	 */
	public MemorizingTrustManager(Context m) {

		init(m);
		this.appTrustManager = getTrustManager(appKeyStore);
		this.defaultTrustManager = getTrustManager(null);
	}

	private void init(Context m) {

		context = m;
		masterHandler = new Handler(m.getMainLooper());
		notificationManager = (NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE);

		keyStoreStorage = m.getSharedPreferences(KEYSTORE_NAME, Context.MODE_PRIVATE);
		appKeyStore = loadAppKeyStore();
	}


	/**
	 * Returns a X509TrustManager list containing a new instance of
	 * TrustManagerFactory.
	 * <p>
	 * This function is meant for convenience only. You can use it
	 * as follows to integrate TrustManagerFactory for HTTPS sockets:
	 *
	 * <pre>
	 *     SSLContext sc = SSLContext.getInstance("TLS");
	 *     sc.init(null, MemorizingTrustManager.getInstanceList(this),
	 *         new java.security.SecureRandom());
	 *     HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());
	 * </pre>
	 *
	 * @param c Activity or Service to show the Dialog / Notification
	 */
	public static X509TrustManager[] getInstanceList(Context c) {

		return new X509TrustManager[]{new MemorizingTrustManager(c)};
	}

	/**
	 * Binds an Activity to the MTM for displaying the query dialog.
	 * <p>
	 * This is useful if your connection is run from a service that is
	 * triggered by user interaction -- in such cases the activity is
	 * visible and the user tends to ignore the service notification.
	 * <p>
	 * You should never have a hidden activity bound to MTM! Use this
	 * function in onResume() and @see unbindDisplayActivity in onPause().
	 *
	 * @param act Activity to be bound
	 */
	private void bindDisplayActivity(Activity act) {

		foregroundAct = act;
	}

	/**
	 * Removes an Activity from the MTM display stack.
	 * <p>
	 * Always call this function when the Activity added with
	 * {@link #bindDisplayActivity(Activity)} is hidden.
	 *
	 * @param act Activity to be unbound
	 */
	public void unbindDisplayActivity(Activity act) {
		// do not remove if it was overridden by a different activity
		if(foregroundAct == act) {
			foregroundAct = null;
		}
	}

	/**
	 * Get a list of all certificate aliases stored in MTM.
	 *
	 * @return an {@link Enumeration} of all certificates
	 */
	private Enumeration<String> getCertificates() {

		try {
			return appKeyStore.aliases();
		}
		catch(KeyStoreException e) {
			// this should never happen, however...
			throw new RuntimeException(e);
		}
	}

	/**
	 * Get a certificate for a given alias.
	 *
	 * @param alias the certificate's alias as returned by {@link #getCertificates()}.
	 * @return the certificate associated with the alias or <tt>null</tt> if none found.
	 */
	public Certificate getCertificate(String alias) {

		try {
			return appKeyStore.getCertificate(alias);
		}
		catch(KeyStoreException e) {
			// this should never happen, however...
			throw new RuntimeException(e);
		}
	}

	/**
	 * Removes the given certificate from MTMs key store.
	 *
	 * <p>
	 * <b>WARNING</b>: this does not immediately invalidate the certificate. It is
	 * well possible that (a) data is transmitted over still existing connections or
	 * (b) new connections are created using TLS renegotiation, without a new cert
	 * check.
	 * </p>
	 *
	 * @param alias the certificate's alias as returned by {@link #getCertificates()}.
	 * @throws KeyStoreException if the certificate could not be deleted.
	 */
	public void deleteCertificate(String alias) throws KeyStoreException {

		appKeyStore.deleteEntry(alias);
		keyStoreUpdated();
	}

	/**
	 * Creates a new hostname verifier supporting user interaction.
	 *
	 * <p>This method creates a new {@link HostnameVerifier} that is bound to
	 * the given instance of {@link MemorizingTrustManager}, and leverages an
	 * existing {@link HostnameVerifier}. The returned verifier performs the
	 * following steps, returning as soon as one of them succeeds:
	 * /p>
	 * <ol>
	 * <li>Success, if the wrapped defaultVerifier accepts the certificate.</li>
	 * <li>Success, if the server certificate is stored in the keystore under the given hostname.</li>
	 * <li>Ask the user and return accordingly.</li>
	 * <li>Failure on exception.</li>
	 * </ol>
	 *
	 * @param defaultVerifier the {@link HostnameVerifier} that should perform the actual check
	 * @return a new hostname verifier using the MTM's key store
	 * @throws IllegalArgumentException if the defaultVerifier parameter is null
	 */
	public HostnameVerifier wrapHostnameVerifier(final HostnameVerifier defaultVerifier) {

		if(defaultVerifier == null) {
			throw new IllegalArgumentException("The default verifier may not be null");
		}

		return new MemorizingHostnameVerifier(defaultVerifier);
	}

	private X509TrustManager getTrustManager(KeyStore ks) {

		try {
			TrustManagerFactory tmf = TrustManagerFactory.getInstance("X509");
			tmf.init(ks);
			for(TrustManager t : tmf.getTrustManagers()) {
				if(t instanceof X509TrustManager) {
					return (X509TrustManager) t;
				}
			}
		}
		catch(Exception e) {
			e.printStackTrace();
		}

		return null;
	}

	private KeyStore loadAppKeyStore() {

		KeyStore keyStore;

		try {
			keyStore = KeyStore.getInstance(KeyStore.getDefaultType());
		}
		catch(KeyStoreException e) {
			e.printStackTrace();
			return null;
		}

		try {
			keyStore.load(null, null);
		}
		catch(NoSuchAlgorithmException | CertificateException | IOException e) {
			e.printStackTrace();
		}

		String keystore = keyStoreStorage.getString(KEYSTORE_KEY, null);

		if(keystore != null) {
			ByteArrayInputStream inputStream = new ByteArrayInputStream(Base64.decode(keystore, Base64.DEFAULT));

			try {
				keyStore.load(inputStream, "MTM".toCharArray());
				inputStream.close();
			}
			catch(Exception e) {
				e.printStackTrace();
			}
		}

		return keyStore;
	}

	private void storeCert(String alias, Certificate cert) {

		try {
			appKeyStore.setCertificateEntry(alias, cert);
		}
		catch(KeyStoreException e) {
			e.printStackTrace();
			return;
		}

		keyStoreUpdated();
	}

	private void storeCert(X509Certificate cert) {

		storeCert(cert.getSubjectDN().toString(), cert);
	}

	private void keyStoreUpdated() {
		// reload appTrustManager
		appTrustManager = getTrustManager(appKeyStore);

		// store KeyStore to shared preferences
		ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();

		try {
			appKeyStore.store(byteArrayOutputStream, "MTM".toCharArray());
			byteArrayOutputStream.flush();
			byteArrayOutputStream.close();

			keyStoreStorage.edit().putString(KEYSTORE_KEY, Base64.encodeToString(byteArrayOutputStream.toByteArray(), Base64.DEFAULT)).apply();
		}
		catch(Exception e) {
			e.printStackTrace();
		}
	}

	// if the certificate is stored in the app key store, it is considered "known"
	private boolean isCertKnown(X509Certificate cert) {

		try {
			return appKeyStore.getCertificateAlias(cert) != null;
		}
		catch(KeyStoreException e) {
			return false;
		}
	}

	private static boolean isExpiredException(Throwable e) {

		do {
			if(e instanceof CertificateExpiredException) {
				return true;
			}
			e = e.getCause();
		} while(e != null);
		return false;
	}

	private static boolean isPathException(Throwable e) {

		do {
			if(e instanceof CertPathValidatorException) {
				return true;
			}
			e = e.getCause();
		} while(e != null);
		return false;
	}

	private void checkCertTrusted(X509Certificate[] chain, String authType, boolean isServer) throws CertificateException {

		LOGGER.log(Level.FINE, "checkCertTrusted(" + Arrays.toString(chain) + ", " + authType + ", " + isServer + ")");

		try {
			LOGGER.log(Level.FINE, "checkCertTrusted: trying appTrustManager");
			if(isServer) {
				appTrustManager.checkServerTrusted(chain, authType);
			}
			else {
				appTrustManager.checkClientTrusted(chain, authType);
			}
		}
		catch(CertificateException ae) {
			LOGGER.log(Level.FINER, "checkCertTrusted: appTrustManager did not verify certificate. Will fall back to secondary verification mechanisms (if any).", ae);
			// if the cert is stored in our appTrustManager, we ignore expiredness
			if(isExpiredException(ae)) {
				LOGGER.log(Level.INFO, "checkCertTrusted: accepting expired certificate from keystore");
				return;
			}
			if(isCertKnown(chain[0])) {
				LOGGER.log(Level.INFO, "checkCertTrusted: accepting cert already stored in keystore");
				return;
			}
			try {
				if(defaultTrustManager == null) {
					LOGGER.fine("No defaultTrustManager set. Verification failed, throwing " + ae);
					throw ae;
				}
				LOGGER.log(Level.FINE, "checkCertTrusted: trying defaultTrustManager");
				if(isServer) {
					defaultTrustManager.checkServerTrusted(chain, authType);
				}
				else {
					defaultTrustManager.checkClientTrusted(chain, authType);
				}
			}
			catch(CertificateException e) {
				LOGGER.log(Level.FINER, "checkCertTrusted: defaultTrustManager failed", e);
				interactCert(chain, authType, e);
			}
		}
	}

	public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {

		checkCertTrusted(chain, authType, false);
	}

	public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {

		checkCertTrusted(chain, authType, true);
	}

	public X509Certificate[] getAcceptedIssuers() {

		return defaultTrustManager.getAcceptedIssuers();
	}

	private static int createDecisionId(MTMDecision d) {

		int myId;
		synchronized(openDecisions) {
			myId = decisionId;
			openDecisions.put(myId, d);
			decisionId += 1;
		}
		return myId;
	}

	private static String hexString(byte[] data) {

		StringBuilder si = new StringBuilder();
		for(int i = 0; i < data.length; i++) {
			si.append(String.format("%02x", data[i]));
			if(i < data.length - 1) {
				si.append(":");
			}
		}
		return si.toString();
	}

	private static String certHash(final X509Certificate cert, String digest) {

		try {
			MessageDigest md = MessageDigest.getInstance(digest);
			md.update(cert.getEncoded());
			return hexString(md.digest());
		}
		catch(CertificateEncodingException | NoSuchAlgorithmException e) {
			return e.getMessage();
		}
	}

	private static void certDetails(StringBuilder si, X509Certificate c) {

		SimpleDateFormat validityDateFormater = new SimpleDateFormat("yyyy-MM-dd", Locale.getDefault());
		si.append("\n");
		si.append(c.getSubjectDN().toString());
		si.append("\n");
		si.append(validityDateFormater.format(c.getNotBefore()));
		si.append(" - ");
		si.append(validityDateFormater.format(c.getNotAfter()));
		si.append("\nSHA-256: ");
		si.append(certHash(c, "SHA-256"));
		si.append("\nSHA-1: ");
		si.append(certHash(c, "SHA-1"));
		si.append("\nSigned by: ");
		si.append(c.getIssuerDN().toString());
		si.append("\n");
	}

	private String certChainMessage(final X509Certificate[] chain, CertificateException cause) {

		Throwable e = cause;
		StringBuilder si = new StringBuilder();

		if(isPathException(e)) {
			si.append(context.getString(R.string.mtm_trust_anchor));
		}
		else if(isExpiredException(e)) {
			si.append(context.getString(R.string.mtm_cert_expired));
		}
		else {
			// get to the cause
			while(e.getCause() != null)
				e = e.getCause();
			si.append(e.getLocalizedMessage());
		}

		si.append("\n\n");
		si.append(context.getString(R.string.mtm_connect_anyway));
		si.append("\n\n");
		si.append(context.getString(R.string.mtm_cert_details));
		for(X509Certificate c : chain) {
			certDetails(si, c);
		}
		return si.toString();
	}

	private String hostNameMessage(X509Certificate cert, String hostname) {

		StringBuilder si = new StringBuilder();

		si.append(context.getString(R.string.mtm_hostname_mismatch, hostname));
		si.append("\n\n");

		try {
			Collection<List<?>> sans = cert.getSubjectAlternativeNames();
			if(sans == null) {
				si.append(cert.getSubjectDN());
				si.append("\n");
			}
			else {
				for(List<?> altName : sans) {
					Object name = altName.get(1);
					if(name instanceof String) {
						si.append("[");
						si.append(altName.get(0));
						si.append("] ");
						si.append(name);
						si.append("\n");
					}
				}
			}
		}
		catch(CertificateParsingException e) {
			e.printStackTrace();
			si.append("<Parsing error: ");
			si.append(e.getLocalizedMessage());
			si.append(">\n");
		}

		si.append("\n");
		si.append(context.getString(R.string.mtm_connect_anyway));
		si.append("\n\n");
		si.append(context.getString(R.string.mtm_cert_details));
		certDetails(si, cert);
		return si.toString();
	}

	/**
	 * Reflectively call
	 * <code>Notification.setLatestEventInfo(Context, CharSequence, CharSequence, PendingIntent)</code>
	 * since it was remove in Android API level 23.
	 */
	private static void setLatestEventInfoReflective(Notification notification, Context context, CharSequence mtmNotification, CharSequence certName, PendingIntent call) {

		Method setLatestEventInfo;

		try {
			setLatestEventInfo = notification.getClass().getMethod("setLatestEventInfo", Context.class, CharSequence.class, CharSequence.class, PendingIntent.class);
		}
		catch(NoSuchMethodException e) {
			throw new IllegalStateException(e);
		}

		try {
			setLatestEventInfo.invoke(notification, context, mtmNotification, certName, call);
		}
		catch(IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
			throw new IllegalStateException(e);
		}
	}

	@TargetApi(Build.VERSION_CODES.HONEYCOMB)
	private void startActivityNotification(Intent intent, int decisionId, String certName) {

		final PendingIntent call = PendingIntent.getActivity(context, 0, intent, 0);
		final String mtmNotification = context.getString(R.string.mtm_notification);

		NotificationCompat.Builder builder = new NotificationCompat.Builder(context, "ssl").setSmallIcon(android.R.drawable.ic_lock_lock).setContentTitle(mtmNotification).setContentText(certName).setTicker(certName).setContentIntent(call).setAutoCancel(true).setPriority(NotificationCompat.PRIORITY_HIGH);

		notificationManager.notify(NOTIFICATION_ID + decisionId, builder.build());
	}

	/**
	 * Returns the top-most entry of the activity stack.
	 *
	 * @return the Context of the currently bound UI or the master context if none is bound
	 */
	Context getUI() {

		return (foregroundAct != null) ? foregroundAct : context;
	}

	private int interact(final String message, final int titleId) {
		/* prepare the MTMDecision blocker object */
		MTMDecision choice = new MTMDecision();
		final int myId = createDecisionId(choice);

		masterHandler.post(new Runnable() {

			public void run() {

				Intent intent = new Intent(context, MemorizingActivity.class);
				intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
				intent.setData(Uri.parse(MemorizingTrustManager.class.getName() + "/" + myId));
				intent.putExtra(DECISION_INTENT_ID, myId);
				intent.putExtra(DECISION_INTENT_CERT, message);
				intent.putExtra(DECISION_TITLE_ID, titleId);

				// we try to directly start the activity and fall back to
				// making a notification. If no foreground activity is set
				// (foregroundAct==null) or if the app developer set an
				// invalid / expired activity, the catch-all fallback is
				// deployed.
				try {
					foregroundAct.startActivity(intent);
				}
				catch(Exception e) {
					startActivityNotification(intent, myId, message);
				}
			}
		});

		try {
			synchronized(choice) {
				choice.wait();
			}
		}
		catch(InterruptedException e) {
			e.printStackTrace();
		}

		return choice.state;
	}

	private void interactCert(final X509Certificate[] chain, String authType, CertificateException cause) throws CertificateException {

		switch(interact(certChainMessage(chain, cause), R.string.mtm_accept_cert)) {
			case MTMDecision.DECISION_ALWAYS:
				storeCert(chain[0]); // only store the server cert, not the whole chain
			case MTMDecision.DECISION_ONCE:
				break;
			default:
				throw (cause);
		}
	}

	private boolean interactHostname(X509Certificate cert, String hostname) {

		switch(interact(hostNameMessage(cert, hostname), R.string.mtm_accept_servername)) {
			case MTMDecision.DECISION_ALWAYS:
				storeCert(hostname, cert);
			case MTMDecision.DECISION_ONCE:
				return true;
			default:
				return false;
		}
	}

	static void interactResult(int decisionId, int choice) {

		MTMDecision d;

		synchronized(openDecisions) {
			d = openDecisions.get(decisionId);
			openDecisions.remove(decisionId);
		}

		if(d == null) {
			return;
		}

		synchronized(d) {
			d.state = choice;
			d.notify();
		}
	}

	class MemorizingHostnameVerifier implements HostnameVerifier {

		private HostnameVerifier defaultVerifier;

		MemorizingHostnameVerifier(HostnameVerifier wrapped) {

			defaultVerifier = wrapped;
		}

		@Override
		public boolean verify(String hostname, SSLSession session) {
			// if the default verifier accepts the hostname, we are done
			if(defaultVerifier.verify(hostname, session)) {
				return true;
			}

			// otherwise, we check if the hostname is an alias for this cert in our keystore
			try {
				X509Certificate cert = (X509Certificate) session.getPeerCertificates()[0];

				if(cert.equals(appKeyStore.getCertificate(hostname.toLowerCase(Locale.US)))) {
					return true;
				}
				else {
					return interactHostname(cert, hostname);
				}
			}
			catch(Exception e) {
				e.printStackTrace();
				return false;
			}
		}

	}

}