src/eu/alefzero/webdav/WebdavClient.java

   1 /* ownCloud Android client application
   2  *   Copyright (C) 2011  Bartek Przybylski
   3  *
   4  *   This program is free software: you can redistribute it and/or modify
   5  *   it under the terms of the GNU General Public License as published by
   6  *   the Free Software Foundation, either version 3 of the License, or
   7  *   (at your option) any later version.
   8  *
   9  *   This program is distributed in the hope that it will be useful,
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *   GNU General Public License for more details.
  13  *
  14  *   You should have received a copy of the GNU General Public License
  15  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  *
  17  */
  18 package eu.alefzero.webdav;
  19
  20 import java.io.BufferedInputStream;
  21 import java.io.File;
  22 import java.io.FileOutputStream;
  23 import java.io.IOException;
  24
  25 import org.apache.commons.httpclient.Credentials;
  26 import org.apache.commons.httpclient.HttpClient;
  27 import org.apache.commons.httpclient.HttpException;
  28 import org.apache.commons.httpclient.HttpMethodBase;
  29 import org.apache.commons.httpclient.HttpVersion;
  30 import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager;
  31 import org.apache.commons.httpclient.UsernamePasswordCredentials;
  32 import org.apache.commons.httpclient.auth.AuthScope;
  33 import org.apache.commons.httpclient.methods.GetMethod;
  34 import org.apache.commons.httpclient.methods.HeadMethod;
  35 import org.apache.commons.httpclient.methods.PutMethod;
  36 import org.apache.commons.httpclient.params.HttpMethodParams;
  37 import org.apache.commons.httpclient.protocol.Protocol;
  38 import org.apache.http.HttpStatus;
  39 import org.apache.http.params.CoreProtocolPNames;
  40 import org.apache.jackrabbit.webdav.client.methods.DavMethod;
  41 import org.apache.jackrabbit.webdav.client.methods.DeleteMethod;
  42 import org.apache.jackrabbit.webdav.client.methods.MkColMethod;
  43
  44 import android.accounts.Account;
  45 import android.accounts.AccountManager;
  46 import android.content.Context;
  47 import android.net.Uri;
  48 import android.util.Log;
  49 import eu.alefzero.owncloud.AccountUtils;
  50 import eu.alefzero.owncloud.authenticator.AccountAuthenticator;
  51 import eu.alefzero.owncloud.authenticator.EasySSLSocketFactory;
  52 import eu.alefzero.owncloud.files.interfaces.OnDatatransferProgressListener;
  53 import eu.alefzero.owncloud.utils.OwnCloudVersion;
  54
  55 public class WebdavClient extends HttpClient {
  56     private Uri mUri;
  57     private Credentials mCredentials;
  58     final private static String TAG = "WebdavClient";
  59     private static final String USER_AGENT = "Android-ownCloud";
  60
  61     /** Default timeout for waiting data from the server: 10 seconds */
  62     public static final int DEFAULT_DATA_TIMEOUT = 10000;
  63
  64     /** Default timeout for establishing a connection: infinite */
  65     public static final int DEFAULT_CONNECTION_TIMEOUT = 0;
  66
  67     private OnDatatransferProgressListener mDataTransferListener;
  68     static private MultiThreadedHttpConnectionManager mConnManager = null;
  69
  70     static public MultiThreadedHttpConnectionManager getMultiThreadedConnManager() {
  71         if (mConnManager == null) {
  72             mConnManager = new MultiThreadedHttpConnectionManager();
  73             mConnManager.setMaxConnectionsPerHost(5);
  74             mConnManager.setMaxTotalConnections(5);
  75         }
  76         return mConnManager;
  77     }
  78
  79     /**
  80      * Creates a WebdavClient setup for the current account
  81      * @param account The client accout
  82      * @param context The application context
  83      * @return
  84      */
  85     public WebdavClient (Account account, Context context) {
  86         setDefaultTimeouts();
  87
  88         OwnCloudVersion ownCloudVersion = new OwnCloudVersion(AccountManager.get(context).getUserData(account,
  89                 AccountAuthenticator.KEY_OC_VERSION));
  90         String baseUrl = AccountManager.get(context).getUserData(account, AccountAuthenticator.KEY_OC_BASE_URL);
  91         String webDavPath = AccountUtils.getWebdavPath(ownCloudVersion);
  92         String username = account.name.substring(0, account.name.lastIndexOf('@'));
  93         String password = AccountManager.get(context).getPassword(account);
  94
  95         mUri = Uri.parse(baseUrl + webDavPath);
  96         Log.e("ASD", ""+username);
  97         setCredentials(username, password);
  98     }
  99
 100     public WebdavClient() {
 101         super(getMultiThreadedConnManager());
 102
 103         setDefaultTimeouts();
 104
 105         getParams().setParameter(HttpMethodParams.USER_AGENT, USER_AGENT);
 106         getParams().setParameter(CoreProtocolPNames.PROTOCOL_VERSION, HttpVersion.HTTP_1_1);
 107         allowSelfsignedCertificates();
 108     }
 109
 110     public void setCredentials(String username, String password) {
 111         getParams().setAuthenticationPreemptive(true);
 112         getState().setCredentials(AuthScope.ANY,
 113                 getCredentials(username, password));
 114     }
 115
 116     private Credentials getCredentials(String username, String password) {
 117         if (mCredentials == null)
 118             mCredentials = new UsernamePasswordCredentials(username, password);
 119         return mCredentials;
 120     }
 121
 122     /**
 123      * Sets the connection and wait-for-data timeouts to be applied by default.
 124      */
 125     private void setDefaultTimeouts() {
 126         getParams().setSoTimeout(DEFAULT_DATA_TIMEOUT);
 127         getHttpConnectionManager().getParams().setConnectionTimeout(DEFAULT_CONNECTION_TIMEOUT);
 128     }
 129
 130     public void allowSelfsignedCertificates() {
 131         // https
 132         Protocol.registerProtocol("https", new Protocol("https",
 133                 new EasySSLSocketFactory(), 443));
 134     }
 135
 136     /**
 137      * Downloads a file in remoteFilepath to the local targetPath.
 138      *
 139      * @param remoteFilepath    Path to the file in the remote server, URL DECODED.
 140      * @param targetFile        Local path to save the downloaded file.
 141      * @return                  'True' when the file is successfully downloaded.
 142      */
 143     public boolean downloadFile(String remoteFilepath, File targetFile) {
 144         boolean ret = false;
 145         boolean caughtException = false;
 146         GetMethod get = new GetMethod(mUri.toString() + WebdavUtils.encodePath(remoteFilepath));
 147
 148         // get.setHeader("Host", mUri.getHost());
 149         // get.setHeader("User-Agent", "Android-ownCloud");
 150
 151         int status = -1;
 152         try {
 153             status = executeMethod(get);
 154             if (status == HttpStatus.SC_OK) {
 155                 targetFile.createNewFile();
 156                 BufferedInputStream bis = new BufferedInputStream(
 157                         get.getResponseBodyAsStream());
 158                 FileOutputStream fos = new FileOutputStream(targetFile);
 159
 160                 byte[] bytes = new byte[4096];
 161                 int readResult;
 162                 while ((readResult = bis.read(bytes)) != -1) {
 163                     if (mDataTransferListener != null)
 164                         mDataTransferListener.transferProgress(readResult);
 165                     fos.write(bytes, 0, readResult);
 166                 }
 167                 ret = true;
 168             }
 169
 170         } catch (HttpException e) {
 171             Log.e(TAG, "HTTP exception downloading " + remoteFilepath, e);
 172             caughtException = true;
 173
 174         } catch (IOException e) {
 175             Log.e(TAG, "I/O exception downloading " + remoteFilepath, e);
 176             caughtException = true;
 177
 178         } catch (Exception e) {
 179             Log.e(TAG, "Unexpected exception downloading " + remoteFilepath, e);
 180             caughtException = true;
 181
 182         } finally {
 183             if (!ret) {
 184                 if (!caughtException) {
 185                     Log.e(TAG, "Download of " + remoteFilepath + " to " + targetFile + " failed with HTTP status " + status);
 186                 }
 187                 if (targetFile.exists()) {
 188                     targetFile.delete();
 189                 }
 190             }
 191         }
 192         return ret;
 193     }
 194
 195     /**
 196      * Deletes a remote file via webdav
 197      * @param remoteFilePath       Remote file path of the file to delete, in URL DECODED format.
 198      * @return
 199      */
 200     public boolean deleteFile(String remoteFilePath){
 201         DavMethod delete = new DeleteMethod(mUri.toString() + WebdavUtils.encodePath(remoteFilePath));
 202         try {
 203             executeMethod(delete);
 204         }  catch (Throwable e) {
 205             Log.e(TAG, "Deleting failed with error: " + e.getMessage(), e);
 206             return false;
 207         }
 208         return true;
 209     }
 210
 211     public void setDataTransferProgressListener(OnDatatransferProgressListener listener) {
 212         mDataTransferListener = listener;
 213     }
 214
 215     /**
 216      * Creates or update a file in the remote server with the contents of a local file.
 217      *
 218      *
 219      * @param localFile         Path to the local file to upload.
 220      * @param remoteTarget      Remote path to the file to create or update, URL DECODED
 221      * @param contentType       MIME type of the file.
 222      * @return                  'True' then the upload was successfully completed
 223      */
 224     public boolean putFile(String localFile, String remoteTarget,
 225             String contentType) {
 226         boolean result = false;
 227
 228         try {
 229             Log.e("ASD", contentType + "");
 230             File f = new File(localFile);
 231             FileRequestEntity entity = new FileRequestEntity(f, contentType);
 232             entity.setOnDatatransferProgressListener(mDataTransferListener);
 233             Log.e("ASD", f.exists() + " " + entity.getContentLength());
 234             PutMethod put = new PutMethod(mUri.toString() + WebdavUtils.encodePath(remoteTarget));
 235             put.setRequestEntity(entity);
 236             Log.d(TAG, "" + put.getURI().toString());
 237             int status = executeMethod(put, 0);
 238             Log.d(TAG, "PUT method return with status " + status);
 239
 240             if (status == HttpStatus.SC_OK || status == HttpStatus.SC_CREATED || status == HttpStatus.SC_NO_CONTENT) {
 241                 result = true;
 242                 Log.i(TAG, "Uploading, done");
 243             }
 244
 245         } catch (final Exception e) {
 246             Log.i(TAG, "" + e.getMessage());
 247             result = false;
 248         }
 249
 250         return result;
 251     }
 252
 253     /**
 254      * Tries to log in to the given WedDavURI, with the given credentials
 255      * @param uri To test
 256      * @param username Username to check
 257      * @param password Password to verify
 258      * @return A {@link HttpStatus}-Code of the result. SC_OK is good.
 259      */
 260     public static int tryToLogin(Uri uri, String username, String password) {
 261         int returnCode = 0;
 262         WebdavClient client = new WebdavClient();
 263         client.setCredentials(username, password);
 264         HeadMethod head = new HeadMethod(uri.toString());
 265         try {
 266             returnCode = client.executeMethod(head);
 267         } catch (HttpException e) {
 268             Log.e(TAG, "HTTP exception trying to login at " + uri.getEncodedPath(), e);
 269         } catch (IOException e) {
 270             Log.e(TAG, "I/O exception trying to login at " + uri.getEncodedPath(), e);
 271         } catch (Exception e) {
 272             Log.e(TAG, "Unexpected exception trying to login at " + uri.getEncodedPath(), e);
 273         }
 274         return returnCode;
 275     }
 276
 277     /**
 278      * Creates a remote directory with the received path.
 279      *
 280      * @param path      Path of the directory to create, URL DECODED
 281      * @return          'True' when the directory is successfully created
 282      */
 283     public boolean createDirectory(String path) {
 284         try {
 285             MkColMethod mkcol = new MkColMethod(mUri.toString() + WebdavUtils.encodePath(path));
 286             int status = executeMethod(mkcol);
 287             Log.d(TAG, "Status returned " + status);
 288             Log.d(TAG, "uri: " + mkcol.getURI().toString());
 289             Log.i(TAG, "Creating dir completed");
 290         } catch (final Exception e) {
 291             e.printStackTrace();
 292             return false;
 293         }
 294         return true;
 295     }
 296
 297
 298     /**
 299      * Check if a file exists in the OC server
 300      *
 301      * @return      'Boolean.TRUE' if the file exists; 'Boolean.FALSE' it doesn't exist; NULL if couldn't be checked
 302      */
 303     public Boolean existsFile(String path) {
 304         try {
 305             HeadMethod head = new HeadMethod(mUri.toString() + WebdavUtils.encodePath(path));
 306             int status = executeMethod(head);
 307             return (status == HttpStatus.SC_OK);
 308         } catch (Exception e) {
 309             e.printStackTrace();
 310             return null;
 311         }
 312     }
 313
 314
 315     /**
 316      * Requests the received method with the received timeout (milliseconds).
 317      *
 318      * Executes the method through the inherited HttpClient.executedMethod(method).
 319      *
 320      * Sets the socket timeout for the HttpMethodBase method received.
 321      *
 322      * @param method    HTTP method request.
 323      * @param timeout   Timeout to set, in milliseconds; <= 0 means infinite.
 324      */
 325     public int executeMethod(HttpMethodBase method, int readTimeout) throws HttpException, IOException {
 326         int oldSoTimeout = getParams().getSoTimeout();
 327         try {
 328             if (readTimeout < 0) {
 329                 readTimeout = 0;
 330             }
 331             HttpMethodParams params = method.getParams();
 332             params.setSoTimeout(readTimeout);
 333             method.setParams(params);               // this should be enough...
 334             getParams().setSoTimeout(readTimeout);  // ... but this is necessary for HTTPS
 335             return executeMethod(method);
 336         } finally {
 337             getParams().setSoTimeout(oldSoTimeout);
 338         }
 339     }
 340 }