netwerk/cache/src/nsDeleteDir.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent: */
   3 /* ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is mozilla.org code.
  17  *
  18  * The Initial Developer of the Original Code is Google Inc.
  19  * Portions created by the Initial Developer are Copyright (C) 2005
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *  Darin Fisher <darin@meer.net>
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either the GNU General Public License Version 2 or later (the "GPL"), or
  27  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38
  39 #include "nsDeleteDir.h"
  40 #include "nsIFile.h"
  41 #include "nsString.h"
  42 #include "prthread.h"
  43
  44 static void DeleteDirThreadFunc(void *arg)
  45 {
  46   nsIFile *dir = static_cast<nsIFile *>(arg);
  47   dir->Remove(PR_TRUE);
  48   NS_RELEASE(dir);
  49 }
  50
  51 nsresult DeleteDir(nsIFile *dirIn, PRBool moveToTrash, PRBool sync)
  52 {
  53   nsresult rv;
  54   nsCOMPtr<nsIFile> trash, dir;
  55
  56   // Need to make a clone of this since we don't want to modify the input
  57   // file object.
  58   rv = dirIn->Clone(getter_AddRefs(dir));
  59   if (NS_FAILED(rv))
  60     return rv;
  61
  62   if (moveToTrash)
  63   {
  64     rv = GetTrashDir(dir, &trash);
  65     if (NS_FAILED(rv))
  66       return rv;
  67
  68     nsCOMPtr<nsIFile> subDir;
  69     rv = trash->Clone(getter_AddRefs(subDir));
  70     if (NS_FAILED(rv))
  71       return rv;
  72
  73     rv = subDir->AppendNative(NS_LITERAL_CSTRING("Trash"));
  74     if (NS_FAILED(rv))
  75       return rv;
  76
  77     rv = subDir->CreateUnique(nsIFile::DIRECTORY_TYPE, 0700);
  78     if (NS_FAILED(rv))
  79       return rv;
  80
  81     rv = dir->MoveToNative(subDir, EmptyCString());
  82     if (NS_FAILED(rv))
  83       return rv;
  84   }
  85   else
  86   {
  87     // we want to pass a clone of the original off to the worker thread.
  88     trash.swap(dir);
  89   }
  90
  91   // Steal ownership of trash directory; let the thread release it.
  92   nsIFile *trashRef = nsnull;
  93   trash.swap(trashRef);
  94
  95   if (sync)
  96   {
  97     DeleteDirThreadFunc(trashRef);
  98   }
  99   else
 100   {
 101     // now, invoke the worker thread
 102     PRThread *thread = PR_CreateThread(PR_USER_THREAD,
 103                                        DeleteDirThreadFunc,
 104                                        trashRef,
 105                                        PR_PRIORITY_LOW,
 106                                        PR_GLOBAL_THREAD,
 107                                        PR_UNJOINABLE_THREAD,
 108                                        0);
 109     if (!thread)
 110       return NS_ERROR_UNEXPECTED;
 111   }
 112
 113   return NS_OK;
 114 }
 115
 116 nsresult GetTrashDir(nsIFile *target, nsCOMPtr<nsIFile> *result)
 117 {
 118   nsresult rv = target->Clone(getter_AddRefs(*result));
 119   if (NS_FAILED(rv))
 120     return rv;
 121
 122   nsCAutoString leaf;
 123   rv = (*result)->GetNativeLeafName(leaf);
 124   if (NS_FAILED(rv))
 125     return rv;
 126   leaf.AppendLiteral(".Trash");
 127
 128   return (*result)->SetNativeLeafName(leaf);
 129 }