| /******************************************************************************* |
| * Copyright (c) 2004, 2007 Composent, Inc. 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: Composent, Inc. - initial API and implementation |
| ******************************************************************************/ |
| package org.eclipse.ecf.provider.filetransfer.retrieve; |
| |
| import java.io.*; |
| import java.net.MalformedURLException; |
| import java.net.URL; |
| import java.util.Map; |
| import org.eclipse.core.net.proxy.IProxyData; |
| import org.eclipse.core.net.proxy.IProxyService; |
| import org.eclipse.core.runtime.*; |
| import org.eclipse.core.runtime.jobs.Job; |
| import org.eclipse.ecf.core.identity.*; |
| import org.eclipse.ecf.core.security.IConnectContext; |
| import org.eclipse.ecf.core.util.Proxy; |
| import org.eclipse.ecf.core.util.ProxyAddress; |
| import org.eclipse.ecf.filetransfer.*; |
| import org.eclipse.ecf.filetransfer.events.*; |
| import org.eclipse.ecf.filetransfer.identity.IFileID; |
| import org.eclipse.ecf.filetransfer.service.IRetrieveFileTransfer; |
| import org.eclipse.ecf.internal.provider.filetransfer.Activator; |
| import org.eclipse.ecf.internal.provider.filetransfer.Messages; |
| import org.eclipse.ecf.provider.filetransfer.identity.FileTransferNamespace; |
| import org.eclipse.osgi.util.NLS; |
| |
| public abstract class AbstractRetrieveFileTransfer implements IIncomingFileTransfer, IRetrieveFileTransfer, IFileTransferPausable { |
| |
| public static final int DEFAULT_BUF_LENGTH = 4096; |
| |
| private static final int FILETRANSFER_ERRORCODE = 1001; |
| |
| protected Job job; |
| |
| protected URL remoteFileURL; |
| |
| protected IFileID remoteFileID; |
| |
| protected IFileTransferListener listener; |
| |
| protected int buff_length = DEFAULT_BUF_LENGTH; |
| |
| protected boolean done = false; |
| |
| protected long bytesReceived = 0; |
| |
| protected InputStream remoteFileContents; |
| |
| protected OutputStream localFileContents; |
| |
| protected boolean closeOutputStream = true; |
| |
| protected Exception exception; |
| |
| protected long fileLength = -1; |
| |
| protected Map options = null; |
| |
| protected boolean paused = false; |
| |
| protected IFileRangeSpecification rangeSpecification = null; |
| |
| protected Proxy proxy; |
| |
| protected IConnectContext connectContext; |
| |
| public void setConnectContextForAuthentication(IConnectContext connectContext) { |
| this.connectContext = connectContext; |
| } |
| |
| /* (non-Javadoc) |
| * @see org.eclipse.ecf.filetransfer.IRetrieveFileTransferContainerAdapter#setProxy(org.eclipse.ecf.core.util.Proxy) |
| */ |
| public void setProxy(Proxy proxy) { |
| this.proxy = proxy; |
| } |
| |
| protected URL getRemoteFileURL() { |
| return remoteFileURL; |
| } |
| |
| protected void setInputStream(InputStream ins) { |
| remoteFileContents = ins; |
| } |
| |
| protected void setOutputStream(OutputStream outs) { |
| localFileContents = outs; |
| } |
| |
| protected void setCloseOutputStream(boolean close) { |
| closeOutputStream = close; |
| } |
| |
| protected void setFileLength(long length) { |
| fileLength = length; |
| } |
| |
| protected Map getOptions() { |
| return options; |
| } |
| |
| public AbstractRetrieveFileTransfer() { |
| // |
| } |
| |
| protected void handleReceivedData(byte[] buf, int bytes, double factor, IProgressMonitor monitor) throws IOException { |
| if (bytes != -1) { |
| bytesReceived += bytes; |
| localFileContents.write(buf, 0, bytes); |
| fireTransferReceiveDataEvent(); |
| monitor.worked((int) Math.round(factor * bytes)); |
| } else |
| done = true; |
| } |
| |
| public class FileTransferJob extends Job { |
| |
| public FileTransferJob(String name) { |
| super(name); |
| } |
| |
| protected IStatus run(IProgressMonitor monitor) { |
| final byte[] buf = new byte[buff_length]; |
| final long totalWork = ((fileLength == -1) ? 100 : fileLength); |
| double factor = (totalWork > Integer.MAX_VALUE) ? (((double) Integer.MAX_VALUE) / ((double) totalWork)) : 1.0; |
| int work = (totalWork > Integer.MAX_VALUE) ? Integer.MAX_VALUE : (int) totalWork; |
| monitor.beginTask(getRemoteFileURL().toString() + Messages.AbstractRetrieveFileTransfer_Progress_Data, work); |
| try { |
| while (!isDone() && !isPaused()) { |
| if (monitor.isCanceled()) |
| throw new UserCancelledException(Messages.AbstractRetrieveFileTransfer_Exception_User_Cancelled); |
| final int bytes = remoteFileContents.read(buf); |
| handleReceivedData(buf, bytes, factor, monitor); |
| } |
| } catch (final Exception e) { |
| exception = e; |
| done = true; |
| } finally { |
| hardClose(); |
| monitor.done(); |
| if (isPaused()) |
| fireTransferReceivePausedEvent(); |
| else |
| fireTransferReceiveDoneEvent(); |
| } |
| return getFinalStatus(exception); |
| } |
| |
| } |
| |
| /* (non-Javadoc) |
| * @see org.eclipse.ecf.core.identity.IIdentifiable#getID() |
| */ |
| public ID getID() { |
| return remoteFileID; |
| } |
| |
| protected IStatus getFinalStatus(Throwable exception1) { |
| if (exception1 == null) |
| return new Status(IStatus.OK, Activator.getDefault().getBundle().getSymbolicName(), 0, Messages.AbstractRetrieveFileTransfer_Status_Transfer_Completed_OK, null); |
| else if (exception1 instanceof UserCancelledException) |
| return new Status(IStatus.CANCEL, Activator.PLUGIN_ID, FILETRANSFER_ERRORCODE, Messages.AbstractRetrieveFileTransfer_Exception_User_Cancelled, exception1); |
| else |
| return new Status(IStatus.ERROR, Activator.PLUGIN_ID, FILETRANSFER_ERRORCODE, Messages.AbstractRetrieveFileTransfer_Status_Transfer_Exception, exception1); |
| } |
| |
| protected void hardClose() { |
| try { |
| if (remoteFileContents != null) |
| remoteFileContents.close(); |
| } catch (final IOException e) { |
| Activator.getDefault().log(new Status(IStatus.ERROR, Activator.PLUGIN_ID, IStatus.ERROR, "hardClose", e)); //$NON-NLS-1$ |
| } |
| try { |
| if (localFileContents != null && closeOutputStream) |
| localFileContents.close(); |
| } catch (final IOException e) { |
| Activator.getDefault().log(new Status(IStatus.ERROR, Activator.PLUGIN_ID, IStatus.ERROR, "hardClose", e)); //$NON-NLS-1$ |
| } |
| job = null; |
| remoteFileContents = null; |
| localFileContents = null; |
| } |
| |
| protected void fireTransferReceivePausedEvent() { |
| listener.handleTransferEvent(new IIncomingFileTransferReceivePausedEvent() { |
| |
| private static final long serialVersionUID = -1317411290525985140L; |
| |
| public IIncomingFileTransfer getSource() { |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| public String toString() { |
| final StringBuffer sb = new StringBuffer("IIncomingFileTransferReceivePausedEvent["); //$NON-NLS-1$ |
| sb.append("bytesReceived=").append(bytesReceived) //$NON-NLS-1$ |
| .append(";fileLength=").append(fileLength).append("]"); //$NON-NLS-1$ //$NON-NLS-2$ |
| return sb.toString(); |
| } |
| }); |
| } |
| |
| protected void fireTransferReceiveDoneEvent() { |
| listener.handleTransferEvent(new IIncomingFileTransferReceiveDoneEvent() { |
| |
| private static final long serialVersionUID = 6925524078226825710L; |
| |
| public IIncomingFileTransfer getSource() { |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| public Exception getException() { |
| return AbstractRetrieveFileTransfer.this.getException(); |
| } |
| |
| public String toString() { |
| final StringBuffer sb = new StringBuffer("IIncomingFileTransferReceiveDoneEvent["); //$NON-NLS-1$ |
| sb.append("bytesReceived=").append(bytesReceived) //$NON-NLS-1$ |
| .append(";fileLength=").append(fileLength).append(";exception=").append(getException()) //$NON-NLS-1$ //$NON-NLS-2$ |
| .append("]"); //$NON-NLS-1$ |
| return sb.toString(); |
| } |
| }); |
| } |
| |
| protected void fireTransferReceiveDataEvent() { |
| listener.handleTransferEvent(new IIncomingFileTransferReceiveDataEvent() { |
| private static final long serialVersionUID = -5656328374614130161L; |
| |
| public IIncomingFileTransfer getSource() { |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| public String toString() { |
| final StringBuffer sb = new StringBuffer("IIncomingFileTransferReceiveDataEvent["); //$NON-NLS-1$ |
| sb.append("bytesReceived=").append(bytesReceived) //$NON-NLS-1$ |
| .append(";fileLength=").append(fileLength) //$NON-NLS-1$ |
| .append("]"); //$NON-NLS-1$ |
| return sb.toString(); |
| } |
| }); |
| } |
| |
| public long getBytesReceived() { |
| return bytesReceived; |
| } |
| |
| public void cancel() { |
| if (isPaused()) { |
| done = true; |
| this.exception = new UserCancelledException(Messages.AbstractRetrieveFileTransfer_Exception_User_Cancelled); |
| fireTransferReceiveDoneEvent(); |
| } else if (job != null) |
| job.cancel(); |
| } |
| |
| public Exception getException() { |
| return exception; |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IFileTransfer#getPercentComplete() |
| */ |
| public double getPercentComplete() { |
| if (fileLength == -1 || fileLength == 0) |
| return fileLength; |
| return ((double) bytesReceived / (double) fileLength); |
| } |
| |
| public boolean isDone() { |
| return done; |
| } |
| |
| public Object getAdapter(Class adapter) { |
| if (adapter == null) |
| return null; |
| if (adapter.isInstance(this)) { |
| return this; |
| } |
| final IAdapterManager adapterManager = Activator.getDefault().getAdapterManager(); |
| return (adapterManager == null) ? null : adapterManager.loadAdapter(this, adapter.getName()); |
| } |
| |
| /** |
| * Open incoming and outgoing streams associated with this file transfer. |
| * Subclasses must implement this method to open input and output streams. |
| * The <code>remoteFileContents</code> and <code>localFileContent</code> |
| * must be non-<code>null</code> after successful completion of the |
| * implementation of this method. |
| * |
| * @throws IncomingFileTransferException |
| */ |
| protected abstract void openStreams() throws IncomingFileTransferException; |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IRetrieveFileTransferContainerAdapter#sendRetrieveRequest(org.eclipse.ecf.filetransfer.identity.IFileID, |
| * org.eclipse.ecf.filetransfer.IFileTransferListener, java.util.Map) |
| */ |
| public void sendRetrieveRequest(final IFileID remoteFileID1, IFileTransferListener transferListener, Map options1) throws IncomingFileTransferException { |
| sendRetrieveRequest(remoteFileID1, null, transferListener, options1); |
| } |
| |
| public Namespace getRetrieveNamespace() { |
| return IDFactory.getDefault().getNamespaceByName(FileTransferNamespace.PROTOCOL); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IFileTransferPausable#isPaused() |
| */ |
| public boolean isPaused() { |
| return paused; |
| } |
| |
| /** |
| * Subclass overridable version of {@link #pause()}. Subclasses must |
| * provide an implementation of this method to support |
| * {@link IFileTransferPausable}. |
| * |
| * @return true if the pause is successful. <code>false</code> otherwise. |
| */ |
| protected abstract boolean doPause(); |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IFileTransferPausable#pause() |
| */ |
| public boolean pause() { |
| return doPause(); |
| } |
| |
| /** |
| * Subclass overridable version of {@link #resume()}. Subclasses must |
| * provide an implementation of this method to support |
| * {@link IFileTransferPausable}. |
| * |
| * @return true if the resume is successful. <code>false</code> otherwise. |
| */ |
| protected abstract boolean doResume(); |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IFileTransferPausable#resume() |
| */ |
| public boolean resume() { |
| return doResume(); |
| } |
| |
| public IFileTransferListener getListener() { |
| return listener; |
| } |
| |
| protected void setupAndScheduleJob() { |
| job = new FileTransferJob(getRemoteFileURL().toString()); |
| job.schedule(); |
| } |
| |
| protected void fireReceiveStartEvent() { |
| listener.handleTransferEvent(new IIncomingFileTransferReceiveStartEvent() { |
| private static final long serialVersionUID = -59096575294481755L; |
| |
| public IFileID getFileID() { |
| return remoteFileID; |
| } |
| |
| public IIncomingFileTransfer receive(File localFileToSave) throws IOException { |
| setOutputStream(new BufferedOutputStream(new FileOutputStream(localFileToSave))); |
| setupAndScheduleJob(); |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| public String toString() { |
| final StringBuffer sb = new StringBuffer("IIncomingFileTransferReceiveStartEvent["); //$NON-NLS-1$ |
| sb.append("isdone=").append(done).append(";"); //$NON-NLS-1$ //$NON-NLS-2$ |
| sb.append("bytesReceived=").append(bytesReceived) //$NON-NLS-1$ |
| .append("]"); //$NON-NLS-1$ |
| return sb.toString(); |
| } |
| |
| public void cancel() { |
| hardClose(); |
| } |
| |
| /** |
| * @param streamToStore |
| * @return incoming file transfer instance. |
| * @throws IOException not thrown in this implementation. |
| */ |
| public IIncomingFileTransfer receive(OutputStream streamToStore) throws IOException { |
| setOutputStream(streamToStore); |
| setCloseOutputStream(false); |
| setupAndScheduleJob(); |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| }); |
| } |
| |
| protected void fireReceiveResumedEvent() { |
| listener.handleTransferEvent(new IIncomingFileTransferReceiveResumedEvent() { |
| |
| private static final long serialVersionUID = 7111739642849612839L; |
| |
| public IFileID getFileID() { |
| return remoteFileID; |
| } |
| |
| public IIncomingFileTransfer receive(File localFileToSave) throws IOException { |
| setOutputStream(new BufferedOutputStream(new FileOutputStream(localFileToSave))); |
| setupAndScheduleJob(); |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| public String toString() { |
| final StringBuffer sb = new StringBuffer("IIncomingFileTransferReceiveResumedEvent["); //$NON-NLS-1$ |
| sb.append("isdone=").append(done).append(";"); //$NON-NLS-1$ //$NON-NLS-2$ |
| sb.append("bytesReceived=").append(bytesReceived) //$NON-NLS-1$ |
| .append("]"); //$NON-NLS-1$ |
| return sb.toString(); |
| } |
| |
| public void cancel() { |
| hardClose(); |
| } |
| |
| /** |
| * @param streamToStore |
| * @return incoming file transfer instance. |
| * @throws IOException not thrown in this implementation. |
| */ |
| public IIncomingFileTransfer receive(OutputStream streamToStore) throws IOException { |
| setOutputStream(streamToStore); |
| setCloseOutputStream(false); |
| setupAndScheduleJob(); |
| return AbstractRetrieveFileTransfer.this; |
| } |
| |
| }); |
| } |
| |
| /* |
| * (non-Javadoc) |
| * |
| * @see org.eclipse.ecf.filetransfer.IIncomingFileTransfer#getFileRangeSpecification() |
| */ |
| public IFileRangeSpecification getFileRangeSpecification() { |
| return rangeSpecification; |
| } |
| |
| /* (non-Javadoc) |
| * @see org.eclipse.ecf.filetransfer.IRetrieveFileTransferContainerAdapter#sendRetrieveRequest(org.eclipse.ecf.filetransfer.identity.IFileID, org.eclipse.ecf.filetransfer.IFileRangeSpecification, org.eclipse.ecf.filetransfer.IFileTransferListener, java.util.Map) |
| */ |
| public void sendRetrieveRequest(IFileID rFileID, IFileRangeSpecification rangeSpec, IFileTransferListener transferListener, Map ops) throws IncomingFileTransferException { |
| Assert.isNotNull(rFileID, Messages.AbstractRetrieveFileTransfer_RemoteFileID_Not_Null); |
| Assert.isNotNull(transferListener, Messages.AbstractRetrieveFileTransfer_TransferListener_Not_Null); |
| this.job = null; |
| this.remoteFileURL = null; |
| this.remoteFileID = rFileID; |
| this.listener = transferListener; |
| this.remoteFileContents = null; |
| this.localFileContents = null; |
| this.closeOutputStream = true; |
| this.done = false; |
| this.exception = null; |
| this.bytesReceived = 0; |
| this.fileLength = -1; |
| this.options = ops; |
| this.paused = false; |
| this.rangeSpecification = rangeSpec; |
| |
| try { |
| this.remoteFileURL = rFileID.getURL(); |
| } catch (final MalformedURLException e) { |
| throw new IncomingFileTransferException(NLS.bind(Messages.AbstractRetrieveFileTransfer_MalformedURLException, rFileID), e); |
| } |
| setupProxies(); |
| openStreams(); |
| } |
| |
| /** |
| * Setup ECF proxy. Subclasses must override this method to do appropriate proxy setup. This method will be called |
| * from within {@link #sendRetrieveRequest(IFileID, IFileTransferListener, Map)} and {@link #sendRetrieveRequest(IFileID, IFileRangeSpecification, IFileTransferListener, Map)}, |
| * prior to the actual call to {@link #openStreams()}. |
| * @param proxy the proxy to be setup. Will not be <code>null</code>. |
| */ |
| protected abstract void setupProxy(Proxy proxy); |
| |
| /** |
| * Select a single proxy from a set of proxies available for the given host. This implementation |
| * selects in the following manner: 1) If proxies provided is null or array of 0 length, null |
| * is returned. If only one proxy is available (array of length 1) then the entry is returned. |
| * If proxies provided is length > 1, then if the type of a proxy in the array matches the given |
| * protocol (e.g. http, https), then the first matching proxy is returned. If the protocol does |
| * not match any of the proxies, then the *first* proxy (i.e. proxies[0]) is returned. Subclasses may |
| * override if desired. |
| * |
| * @param protocol the target protocol (e.g. http, https, scp, etc). Will not be <code>null</code>. |
| * @param proxies the proxies to select from. May be <code>null</code> or array of length 0. |
| * @return proxy data selected from the proxies provided. |
| */ |
| protected IProxyData selectProxyFromProxies(String protocol, IProxyData[] proxies) { |
| if (proxies == null || proxies.length == 0) |
| return null; |
| // If only one proxy is available, then use that |
| if (proxies.length == 1) |
| return proxies[0]; |
| // If more than one proxy is available, then if http/https protocol then look for that |
| // one...if not found then use first |
| if (protocol.equalsIgnoreCase("http")) { //$NON-NLS-1$ |
| for (int i = 0; i < proxies.length; i++) { |
| if (proxies[i].getType().equals(IProxyData.HTTP_PROXY_TYPE)) |
| return proxies[i]; |
| } |
| } else if (protocol.equalsIgnoreCase("https")) { //$NON-NLS-1$ |
| for (int i = 0; i < proxies.length; i++) { |
| if (proxies[i].getType().equals(IProxyData.HTTPS_PROXY_TYPE)) |
| return proxies[i]; |
| } |
| } |
| // If we haven't found it yet, then return the first one. |
| return proxies[0]; |
| } |
| |
| protected void setupProxies() { |
| // If it's been set directly (via ECF API) then this overrides platform settings |
| if (proxy == null) { |
| try { |
| IProxyService proxyService = Activator.getDefault().getProxyService(); |
| // Only do this if platform service exists |
| if (proxyService != null && proxyService.isProxiesEnabled()) { |
| // Setup via proxyService entry |
| URL target = getRemoteFileURL(); |
| final IProxyData[] proxies = proxyService.getProxyDataForHost(target.getHost()); |
| IProxyData selectedProxy = selectProxyFromProxies(target.getProtocol(), proxies); |
| if (selectedProxy != null) { |
| proxy = new Proxy(((selectedProxy.getType().equalsIgnoreCase(IProxyData.SOCKS_PROXY_TYPE)) ? Proxy.Type.SOCKS : Proxy.Type.HTTP), new ProxyAddress(selectedProxy.getHost(), selectedProxy.getPort()), selectedProxy.getUserId(), selectedProxy.getPassword()); |
| } |
| } |
| } catch (Exception e) { |
| // If we don't even have the classes for this (i.e. the org.eclipse.core.net plugin not available) |
| // then we simply log and ignore |
| Activator.logNoProxyWarning(e); |
| } catch (NoClassDefFoundError e) { |
| Activator.logNoProxyWarning(e); |
| } |
| } |
| if (proxy != null) |
| setupProxy(proxy); |
| } |
| |
| } |