tikhomirov@699: /*
tikhomirov@699:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@699:  *  
tikhomirov@699:  * This program is free software; you can redistribute it and/or modify
tikhomirov@699:  * it under the terms of the GNU General Public License as published by
tikhomirov@699:  * the Free Software Foundation; version 2 of the License.
tikhomirov@699:  *
tikhomirov@699:  * This program is distributed in the hope that it will be useful,
tikhomirov@699:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@699:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@699:  * GNU General Public License for more details.
tikhomirov@699:  *
tikhomirov@699:  * For information on how to redistribute this software under
tikhomirov@699:  * the terms of a license other than GNU General Public License
tikhomirov@699:  * contact TMate Software at support@hg4j.com
tikhomirov@699:  */
tikhomirov@699: package org.tmatesoft.hg.internal.remote;
tikhomirov@699: 
tikhomirov@699: import java.io.File;
tikhomirov@699: import java.io.FileInputStream;
tikhomirov@699: import java.io.IOException;
tikhomirov@699: 
tikhomirov@699: import org.tmatesoft.hg.auth.HgAuthFailedException;
tikhomirov@699: import org.tmatesoft.hg.auth.HgAuthMethod;
tikhomirov@699: import org.tmatesoft.hg.auth.HgAuthenticator;
tikhomirov@699: import org.tmatesoft.hg.repo.HgRemoteRepository.RemoteDescriptor;
tikhomirov@699: import org.tmatesoft.hg.util.LogFacility;
tikhomirov@699: import org.tmatesoft.hg.util.LogFacility.Severity;
tikhomirov@699: 
tikhomirov@699: /**
tikhomirov@699:  * 
tikhomirov@699:  * @author Artem Tikhomirov
tikhomirov@699:  * @author TMate Software Ltd.
tikhomirov@699:  */
tikhomirov@699: public class BasicAuthenticator implements HgAuthenticator {
tikhomirov@699: 	private final LogFacility log;
tikhomirov@699: 
tikhomirov@699: 	public BasicAuthenticator(LogFacility logFacility) {
tikhomirov@699: 		log = logFacility;
tikhomirov@699: 	}
tikhomirov@699: 			
tikhomirov@699: 	public void authenticate(RemoteDescriptor rd, HgAuthMethod authMethod) throws HgAuthFailedException {
tikhomirov@699: 		if (authMethod.supportsPublicKey()) {
tikhomirov@699: 			if (tryPlatformDefaultKeyLocations(rd, authMethod)) {
tikhomirov@699: 				return;
tikhomirov@699: 			}
tikhomirov@699: 		}
tikhomirov@699: 		authMethod.noCredentials();
tikhomirov@699: 	}
tikhomirov@699: 
tikhomirov@699: 	// return true is successfully aithenticated
tikhomirov@699: 	protected boolean tryPlatformDefaultKeyLocations(RemoteDescriptor rd, HgAuthMethod authMethod) {
tikhomirov@699: 		final String userHome = System.getProperty("user.home");
tikhomirov@699: 		File sshDir = new File(userHome, ".ssh");
tikhomirov@699: 		if (!sshDir.isDirectory()) {
tikhomirov@699: 			return false;
tikhomirov@699: 		}
tikhomirov@699: 		final String username = System.getProperty("user.name");
tikhomirov@699: 		for (String fn : new String[] { "id_rsa", "id_dsa", "identity"}) {
tikhomirov@699: 			File id = new File(sshDir, fn);
tikhomirov@699: 			if (!id.canRead()) {
tikhomirov@699: 				continue;
tikhomirov@699: 			}
tikhomirov@699: 			try {
tikhomirov@699: 				FileInputStream fis = new FileInputStream(id);
tikhomirov@699: 				authMethod.withPublicKey(username, fis, null);
tikhomirov@699: 				fis.close();
tikhomirov@699: 				return true;
tikhomirov@699: 			} catch (IOException ex) {
tikhomirov@699: 				log.dump(getClass(), Severity.Warn, ex, String.format("Attempting default ssh identity key locations: %s", id));
tikhomirov@699: 				// ignore
tikhomirov@699: 			} catch (HgAuthFailedException ex) {
tikhomirov@699: 				log.dump(getClass(), Severity.Debug, ex, String.format("Attempting default ssh identity key locations: %s", id));
tikhomirov@699: 				// ignore
tikhomirov@699: 			}
tikhomirov@699: 		}
tikhomirov@699: 		return false;
tikhomirov@699: 	}
tikhomirov@699: }