rms/org.eclipse.ptp.rm.smoa.core/src/org/eclipse/ptp/rm/smoa/core/rservices/SMOAConnection.java - ptp/org.eclipse.ptp - Git at Google

 /*******************************************************************************
  * Copyright (c) 2010 Poznan Supercomputing and Networking Center
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v1.0
  * which accompanies this distribution, and is available at
  * http://www.eclipse.org/legal/epl-v10.html
  *
  * Contributors:
  *     Jan Konczak (PSNC) - initial implementation
  ******************************************************************************/

 package org.eclipse.ptp.rm.smoa.core.rservices;

 import java.io.IOException;
 import java.security.GeneralSecurityException;
 import java.util.Map;

 import javax.xml.bind.JAXBException;

 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.ptp.remote.core.IRemoteConnection;
 import org.eclipse.ptp.remote.core.IRemoteConnectionChangeListener;
 import org.eclipse.ptp.remote.core.IRemoteFileManager;
 import org.eclipse.ptp.remote.core.IRemoteServices;
 import org.eclipse.ptp.remote.core.exception.RemoteConnectionException;
 import org.eclipse.ptp.rm.smoa.core.SMOAConfiguration.AuthType;

 import com.smoa.comp.sdk.SMOAFactory;
 import com.smoa.comp.sdk.SMOARsync;
 import com.smoa.comp.sdk.SMOAStaging;
 import com.smoa.comp.sdk.jsdl.JSDL;
 import com.smoa.comp.stubs.staging.FileNotFoundFault;
 import com.smoa.comp.stubs.staging.NotAuthorizedFault;
 import com.smoa.core.sdk.attachments.FileStagingHandler;
 import com.smoa.core.sdk.security.AuthenticationModule;
 import com.smoa.core.sdk.security.anonymous.AnonymousAuthentication;
 import com.smoa.core.sdk.security.gsi.GSIAuthentication;
 import com.smoa.core.sdk.security.ssl.SSLAuthentication;
 import com.smoa.core.sdk.security.wsse.WSSEAuthentication;
 import com.smoa.core.sdk.security.wsse.username.UsernameToken;

 /**
  * Main class for maintaining the real connection with SMOA Computing.
  *
  * Keeps the SMOAFactory, SMOAStaging and SMOARsync objects.
  */
 public class SMOAConnection implements IRemoteConnection {

 	// Keys for map used to transfer connection attributes
 	public static final String TAG_ADDRESS = "address"; //$NON-NLS-1$
 	public static final String TAG_PORT = "port"; //$NON-NLS-1$
 	public static final String TAG_AUTHTYPE = "auth"; //$NON-NLS-1$
 	public static final String TAG_USERNAME = "user"; //$NON-NLS-1$
 	public static final String TAG_PASSWORD = "pass"; //$NON-NLS-1$
 	public static final String TAG_CACERT = "cacert"; //$NON-NLS-1$
 	public static final String TAG_DN = "dn"; //$NON-NLS-1$

 	// parent RemoteServices
 	private final SMOARemoteServices remoteServices;

 	// Connection state
 	private boolean connectionIsOpen = false;

 	// Crucial objects
 	private SMOAFactory besFactory;
 	private SMOAStaging smoaStaging;
 	private SMOARsync rsync;

 	// Connection info
 	private String name = null;
 	private String url = null;
 	private int port = 0;
 	private String resourceManagerName;

 	// Authentication info
 	private AuthType auth;
 	private String username;
 	private String password;
 	private String dn;
 	private String cacert;

 	// Others
 	private IRemoteFileManager fileManager = null;
 	private String workDir = "."; //$NON-NLS-1$
 	private String homeDir = null;
 	private IRemoteConnection fileRemoteConnection = null;

 	public SMOAConnection(SMOARemoteServices remoteServices, String name) {
 		this.remoteServices = remoteServices;
 		this.name = name;
 	}

 	public void addConnectionChangeListener(IRemoteConnectionChangeListener listener) {
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.addConnectionChangeListener(listener);
 		}

 	}

 	public void close() {
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.close();
 		}
 	}

 	public void forwardLocalPort(int localPort, String fwdAddress, int fwdPort) throws RemoteConnectionException {
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.forwardLocalPort(localPort, fwdAddress, fwdPort);
 		}

 	}

 	public int forwardLocalPort(String fwdAddress, int fwdPort, IProgressMonitor monitor) throws RemoteConnectionException {
 		if (fileRemoteConnection != null) {
 			return fileRemoteConnection.forwardLocalPort(fwdAddress, fwdPort, monitor);
 		}
 		return 0;
 	}

 	public void forwardRemotePort(int remotePort, String fwdAddress, int fwdPort) throws RemoteConnectionException {
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.forwardRemotePort(remotePort, fwdAddress, fwdPort);
 		}

 	}

 	public int forwardRemotePort(String fwdAddress, int fwdPort, IProgressMonitor monitor) throws RemoteConnectionException {
 		if (fileRemoteConnection != null) {
 			return fileRemoteConnection.forwardRemotePort(fwdAddress, fwdPort, monitor);
 		}
 		return 0;
 	}

 	public String getAddress() {
 		return url;
 	}

 	public Map<String, String> getAttributes() {
 		// TODO Auto-generated method stub
 		return null;
 	}

 	public Map<String, String> getEnv() {
 		// TODO Auto-generated method stub
 		return null;
 	}

 	public String getEnv(String name) {
 		// TODO Auto-generated method stub
 		return null;
 	}

 	public SMOAFactory getFactory() {
 		return besFactory;
 	}

 	public IRemoteConnection getFileConnection() {
 		return fileRemoteConnection;
 	}

 	public IRemoteFileManager getFileManager() {
 		return fileManager;
 	}

 	public String getHomeDir() {
 		if (homeDir != null) {
 			return homeDir;
 		}

 		if (!isOpen()) {
 			return "."; //$NON-NLS-1$
 		}

 		JSDL jsdl;
 		try {
 			jsdl = smoaStaging.listDirectory(".", null); //$NON-NLS-1$
 		} catch (final FileNotFoundFault e) {
 			throw new RuntimeException();
 		} catch (final NotAuthorizedFault e) {
 			throw new RuntimeException();
 		}
 		homeDir = jsdl.getWorkingDirectory();
 		return homeDir;

 	}

 	public String getName() {
 		return name;
 	}

 	public int getPort() {
 		return port;
 	}

 	public String getProperty(String key) {
 		// TODO Auto-generated method stub
 		return null;
 	}

 	public IRemoteServices getRemoteServices() {
 		return remoteServices;
 	}

 	public String getRMName() {
 		return resourceManagerName;
 	}

 	public SMOARsync getRsync() {
 		return rsync;
 	}

 	public SMOAStaging getSMOAStaging() {
 		return smoaStaging;
 	}

 	public String getUsername() {
 		return username;
 	}

 	public String getWorkingDirectory() {
 		if (workDir.equals(".")) { //$NON-NLS-1$
 			getHomeDir();
 			if (homeDir != null) {
 				workDir = getHomeDir();
 			} else {
 				return "."; //$NON-NLS-1$
 			}
 		}
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.setWorkingDirectory(workDir);
 		}
 		return workDir;
 	}

 	public boolean isOpen() {
 		if (fileRemoteConnection != null && !fileRemoteConnection.isOpen()) {
 			return false;
 		}
 		return connectionIsOpen;
 	}

 	public void open(IProgressMonitor monitor) throws RemoteConnectionException {
 		try {
 			final AnonymousAuthentication anonAuth = new AnonymousAuthentication();
 			final FileStagingHandler fileStagingHandler = new SMOAFileStagingHandler();

 			switch (auth) {
 			case Anonymous:
 				besFactory = new SMOAFactory(url, port, anonAuth, true);
 				smoaStaging = new SMOAStaging(url, port, anonAuth, true, fileStagingHandler);
 				rsync = new SMOARsync(url, port, anonAuth, true);
 				break;
 			case GSI:
 				final GSIAuthentication gsi = new GSIAuthentication(dn != null && !dn.isEmpty() ? dn : null);
 				besFactory = new SMOAFactory(url, port, gsi, true);
 				smoaStaging = new SMOAStaging(url, port, gsi, true, fileStagingHandler);
 				rsync = new SMOARsync(url, port, gsi, true);

 				break;
 			case UsernamePassword: {

 				final WSSEAuthentication wsseAuth = new WSSEAuthentication(new UsernameToken(username, password, false, false));

 				AuthenticationModule am;

 				if (cacert == null) {
 					am = anonAuth;
 				} else {
 					am = new SSLAuthentication(cacert, null, dn != null && !dn.isEmpty() ? dn : null);

 				}

 				rsync = new SMOARsync(url, port, am, wsseAuth);
 				besFactory = new SMOAFactory(url, port, am, wsseAuth);
 				smoaStaging = new SMOAStaging(url, port, am, wsseAuth, true, fileStagingHandler);

 				break;
 			}
 			default:
 				throw new RuntimeException(Messages.SMOAConnection_UnsupportedAuthType);
 			}
 			connectionIsOpen = true;

 		} catch (final JAXBException e) {
 			throw new RemoteConnectionException(e);
 		} catch (final IOException e) {
 			throw new RemoteConnectionException(e);
 		} catch (final GeneralSecurityException e) {
 			throw new RemoteConnectionException(e);
 		}
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.open(monitor);
 		}
 	}

 	public void removeConnectionChangeListener(IRemoteConnectionChangeListener listener) {
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.removeConnectionChangeListener(listener);
 		}

 	}

 	public void setAddress(String address) {
 		if (address == null) {
 			return;
 		}
 		if (address.equals(url)) {
 			return;
 		}
 		if (isOpen()) {
 			throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
 		}
 		url = address;
 	}

 	public void setAuthType(AuthType auth) {
 		if (this.auth == auth) {
 			return;
 		}
 		if (isOpen()) {
 			throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
 		}
 		this.auth = auth;
 	}

 	public void setCaCert(String string) {
 		cacert = string;
 	}

 	public void setDN(String string) {
 		dn = string;
 	}

 	public void setFileConnection(IRemoteConnection fileRemoteConnection) {
 		if (isOpen()) {
 			throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
 		}
 		this.fileRemoteConnection = fileRemoteConnection;
 	}

 	public void setFileManager(IRemoteFileManager fileManager) {
 		this.fileManager = fileManager;
 	}

 	public void setName(String name) {
 		this.name = name;
 	}

 	public void setPort(int port) {
 		if (this.port == port) {
 			return;
 		}
 		if (isOpen()) {
 			throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
 		}
 		this.port = port;
 	}

 	public void setRMName(String rMName) {
 		resourceManagerName = rMName;
 	}

 	public void setUsername(String username) {
 		if (isOpen()) {
 			throw new RuntimeException();
 		}
 		this.username = username;
 	}

 	public void setWorkingDirectory(String path) {
 		workDir = path;
 		if (fileRemoteConnection != null) {
 			fileRemoteConnection.setWorkingDirectory(path);
 		}

 	}

 	public boolean supportsTCPPortForwarding() {
 		if (fileRemoteConnection != null) {
 			return fileRemoteConnection.supportsTCPPortForwarding();
 		}
 		return false;
 	}

 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.eclipse.ptp.remote.core.IRemoteConnection#setAttribute(java.lang.
 	 * String, java.lang.String)
 	 */
 	public void setAttribute(String key, String value) {
 		// TODO Auto-generated method stub

 	}

 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.eclipse.ptp.remote.core.IRemoteConnection#setPassword(java.lang.String
 	 * )
 	 */
 	public void setPassword(String password) {
 		if (isOpen()) {
 			throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
 		}
 		this.password = password;
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2010 Poznan Supercomputing and Networking Center
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License v1.0
	* which accompanies this distribution, and is available at
	* http://www.eclipse.org/legal/epl-v10.html
	*
	* Contributors:
	* Jan Konczak (PSNC) - initial implementation
	******************************************************************************/

	package org.eclipse.ptp.rm.smoa.core.rservices;

	import java.io.IOException;
	import java.security.GeneralSecurityException;
	import java.util.Map;

	import javax.xml.bind.JAXBException;

	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.ptp.remote.core.IRemoteConnection;
	import org.eclipse.ptp.remote.core.IRemoteConnectionChangeListener;
	import org.eclipse.ptp.remote.core.IRemoteFileManager;
	import org.eclipse.ptp.remote.core.IRemoteServices;
	import org.eclipse.ptp.remote.core.exception.RemoteConnectionException;
	import org.eclipse.ptp.rm.smoa.core.SMOAConfiguration.AuthType;

	import com.smoa.comp.sdk.SMOAFactory;
	import com.smoa.comp.sdk.SMOARsync;
	import com.smoa.comp.sdk.SMOAStaging;
	import com.smoa.comp.sdk.jsdl.JSDL;
	import com.smoa.comp.stubs.staging.FileNotFoundFault;
	import com.smoa.comp.stubs.staging.NotAuthorizedFault;
	import com.smoa.core.sdk.attachments.FileStagingHandler;
	import com.smoa.core.sdk.security.AuthenticationModule;
	import com.smoa.core.sdk.security.anonymous.AnonymousAuthentication;
	import com.smoa.core.sdk.security.gsi.GSIAuthentication;
	import com.smoa.core.sdk.security.ssl.SSLAuthentication;
	import com.smoa.core.sdk.security.wsse.WSSEAuthentication;
	import com.smoa.core.sdk.security.wsse.username.UsernameToken;

	/**
	* Main class for maintaining the real connection with SMOA Computing.
	*
	* Keeps the SMOAFactory, SMOAStaging and SMOARsync objects.
	*/
	public class SMOAConnection implements IRemoteConnection {

	// Keys for map used to transfer connection attributes
	public static final String TAG_ADDRESS = "address"; //$NON-NLS-1$
	public static final String TAG_PORT = "port"; //$NON-NLS-1$
	public static final String TAG_AUTHTYPE = "auth"; //$NON-NLS-1$
	public static final String TAG_USERNAME = "user"; //$NON-NLS-1$
	public static final String TAG_PASSWORD = "pass"; //$NON-NLS-1$
	public static final String TAG_CACERT = "cacert"; //$NON-NLS-1$
	public static final String TAG_DN = "dn"; //$NON-NLS-1$

	// parent RemoteServices
	private final SMOARemoteServices remoteServices;

	// Connection state
	private boolean connectionIsOpen = false;

	// Crucial objects
	private SMOAFactory besFactory;
	private SMOAStaging smoaStaging;
	private SMOARsync rsync;

	// Connection info
	private String name = null;
	private String url = null;
	private int port = 0;
	private String resourceManagerName;

	// Authentication info
	private AuthType auth;
	private String username;
	private String password;
	private String dn;
	private String cacert;

	// Others
	private IRemoteFileManager fileManager = null;
	private String workDir = "."; //$NON-NLS-1$
	private String homeDir = null;
	private IRemoteConnection fileRemoteConnection = null;

	public SMOAConnection(SMOARemoteServices remoteServices, String name) {
	this.remoteServices = remoteServices;
	this.name = name;
	}

	public void addConnectionChangeListener(IRemoteConnectionChangeListener listener) {
	if (fileRemoteConnection != null) {
	fileRemoteConnection.addConnectionChangeListener(listener);
	}

	}

	public void close() {
	if (fileRemoteConnection != null) {
	fileRemoteConnection.close();
	}
	}

	public void forwardLocalPort(int localPort, String fwdAddress, int fwdPort) throws RemoteConnectionException {
	if (fileRemoteConnection != null) {
	fileRemoteConnection.forwardLocalPort(localPort, fwdAddress, fwdPort);
	}

	}

	public int forwardLocalPort(String fwdAddress, int fwdPort, IProgressMonitor monitor) throws RemoteConnectionException {
	if (fileRemoteConnection != null) {
	return fileRemoteConnection.forwardLocalPort(fwdAddress, fwdPort, monitor);
	}
	return 0;
	}

	public void forwardRemotePort(int remotePort, String fwdAddress, int fwdPort) throws RemoteConnectionException {
	if (fileRemoteConnection != null) {
	fileRemoteConnection.forwardRemotePort(remotePort, fwdAddress, fwdPort);
	}

	}

	public int forwardRemotePort(String fwdAddress, int fwdPort, IProgressMonitor monitor) throws RemoteConnectionException {
	if (fileRemoteConnection != null) {
	return fileRemoteConnection.forwardRemotePort(fwdAddress, fwdPort, monitor);
	}
	return 0;
	}

	public String getAddress() {
	return url;
	}

	public Map<String, String> getAttributes() {
	// TODO Auto-generated method stub
	return null;
	}

	public Map<String, String> getEnv() {
	// TODO Auto-generated method stub
	return null;
	}

	public String getEnv(String name) {
	// TODO Auto-generated method stub
	return null;
	}

	public SMOAFactory getFactory() {
	return besFactory;
	}

	public IRemoteConnection getFileConnection() {
	return fileRemoteConnection;
	}

	public IRemoteFileManager getFileManager() {
	return fileManager;
	}

	public String getHomeDir() {
	if (homeDir != null) {
	return homeDir;
	}

	if (!isOpen()) {
	return "."; //$NON-NLS-1$
	}

	JSDL jsdl;
	try {
	jsdl = smoaStaging.listDirectory(".", null); //$NON-NLS-1$
	} catch (final FileNotFoundFault e) {
	throw new RuntimeException();
	} catch (final NotAuthorizedFault e) {
	throw new RuntimeException();
	}
	homeDir = jsdl.getWorkingDirectory();
	return homeDir;

	}

	public String getName() {
	return name;
	}

	public int getPort() {
	return port;
	}

	public String getProperty(String key) {
	// TODO Auto-generated method stub
	return null;
	}

	public IRemoteServices getRemoteServices() {
	return remoteServices;
	}

	public String getRMName() {
	return resourceManagerName;
	}

	public SMOARsync getRsync() {
	return rsync;
	}

	public SMOAStaging getSMOAStaging() {
	return smoaStaging;
	}

	public String getUsername() {
	return username;
	}

	public String getWorkingDirectory() {
	if (workDir.equals(".")) { //$NON-NLS-1$
	getHomeDir();
	if (homeDir != null) {
	workDir = getHomeDir();
	} else {
	return "."; //$NON-NLS-1$
	}
	}
	if (fileRemoteConnection != null) {
	fileRemoteConnection.setWorkingDirectory(workDir);
	}
	return workDir;
	}

	public boolean isOpen() {
	if (fileRemoteConnection != null && !fileRemoteConnection.isOpen()) {
	return false;
	}
	return connectionIsOpen;
	}

	public void open(IProgressMonitor monitor) throws RemoteConnectionException {
	try {
	final AnonymousAuthentication anonAuth = new AnonymousAuthentication();
	final FileStagingHandler fileStagingHandler = new SMOAFileStagingHandler();

	switch (auth) {
	case Anonymous:
	besFactory = new SMOAFactory(url, port, anonAuth, true);
	smoaStaging = new SMOAStaging(url, port, anonAuth, true, fileStagingHandler);
	rsync = new SMOARsync(url, port, anonAuth, true);
	break;
	case GSI:
	final GSIAuthentication gsi = new GSIAuthentication(dn != null && !dn.isEmpty() ? dn : null);
	besFactory = new SMOAFactory(url, port, gsi, true);
	smoaStaging = new SMOAStaging(url, port, gsi, true, fileStagingHandler);
	rsync = new SMOARsync(url, port, gsi, true);

	break;
	case UsernamePassword: {

	final WSSEAuthentication wsseAuth = new WSSEAuthentication(new UsernameToken(username, password, false, false));

	AuthenticationModule am;

	if (cacert == null) {
	am = anonAuth;
	} else {
	am = new SSLAuthentication(cacert, null, dn != null && !dn.isEmpty() ? dn : null);

	}

	rsync = new SMOARsync(url, port, am, wsseAuth);
	besFactory = new SMOAFactory(url, port, am, wsseAuth);
	smoaStaging = new SMOAStaging(url, port, am, wsseAuth, true, fileStagingHandler);

	break;
	}
	default:
	throw new RuntimeException(Messages.SMOAConnection_UnsupportedAuthType);
	}
	connectionIsOpen = true;

	} catch (final JAXBException e) {
	throw new RemoteConnectionException(e);
	} catch (final IOException e) {
	throw new RemoteConnectionException(e);
	} catch (final GeneralSecurityException e) {
	throw new RemoteConnectionException(e);
	}
	if (fileRemoteConnection != null) {
	fileRemoteConnection.open(monitor);
	}
	}

	public void removeConnectionChangeListener(IRemoteConnectionChangeListener listener) {
	if (fileRemoteConnection != null) {
	fileRemoteConnection.removeConnectionChangeListener(listener);
	}

	}

	public void setAddress(String address) {
	if (address == null) {
	return;
	}
	if (address.equals(url)) {
	return;
	}
	if (isOpen()) {
	throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
	}
	url = address;
	}

	public void setAuthType(AuthType auth) {
	if (this.auth == auth) {
	return;
	}
	if (isOpen()) {
	throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
	}
	this.auth = auth;
	}

	public void setCaCert(String string) {
	cacert = string;
	}

	public void setDN(String string) {
	dn = string;
	}

	public void setFileConnection(IRemoteConnection fileRemoteConnection) {
	if (isOpen()) {
	throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
	}
	this.fileRemoteConnection = fileRemoteConnection;
	}

	public void setFileManager(IRemoteFileManager fileManager) {
	this.fileManager = fileManager;
	}

	public void setName(String name) {
	this.name = name;
	}

	public void setPort(int port) {
	if (this.port == port) {
	return;
	}
	if (isOpen()) {
	throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
	}
	this.port = port;
	}

	public void setRMName(String rMName) {
	resourceManagerName = rMName;
	}

	public void setUsername(String username) {
	if (isOpen()) {
	throw new RuntimeException();
	}
	this.username = username;
	}

	public void setWorkingDirectory(String path) {
	workDir = path;
	if (fileRemoteConnection != null) {
	fileRemoteConnection.setWorkingDirectory(path);
	}

	}

	public boolean supportsTCPPortForwarding() {
	if (fileRemoteConnection != null) {
	return fileRemoteConnection.supportsTCPPortForwarding();
	}
	return false;
	}

	/*
	* (non-Javadoc)
	*
	* @see
	* org.eclipse.ptp.remote.core.IRemoteConnection#setAttribute(java.lang.
	* String, java.lang.String)
	*/
	public void setAttribute(String key, String value) {
	// TODO Auto-generated method stub

	}

	/*
	* (non-Javadoc)
	*
	* @see
	* org.eclipse.ptp.remote.core.IRemoteConnection#setPassword(java.lang.String
	* )
	*/
	public void setPassword(String password) {
	if (isOpen()) {
	throw new RuntimeException(Messages.SMOAConnection_CannotModifyOpenConnection);
	}
	this.password = password;
	}
	}