Back to index

lightning-sunbird  0.9+nobinonly
nsTempfilePS.cpp
Go to the documentation of this file.
00001 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
00002 /* ex: set tabstop=8 softtabstop=4 shiftwidth=4 expandtab: */
00003 /* ***** BEGIN LICENSE BLOCK *****
00004  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
00005  *
00006  * The contents of this file are subject to the Mozilla Public License Version
00007  * 1.1 (the "License"); you may not use this file except in compliance with
00008  * the License. You may obtain a copy of the License at
00009  * http://www.mozilla.org/MPL/
00010  *
00011  * Software distributed under the License is distributed on an "AS IS" basis,
00012  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
00013  * for the specific language governing rights and limitations under the
00014  * License.
00015  *
00016  * The Original Code is mozilla.org code.
00017  *
00018  * The Initial Developer of the Original Code is
00019  * Kenneth Herron <kherron@newsguy.com>.
00020  * Portions created by the Initial Developer are Copyright (C) 2003
00021  * the Initial Developer. All Rights Reserved.
00022  *
00023  * Contributor(s):
00024  *   Kenneth Herron <kherron@newsguy.com>
00025  *
00026  * Alternatively, the contents of this file may be used under the terms of
00027  * either of the GNU General Public License Version 2 or later (the "GPL"),
00028  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
00029  * in which case the provisions of the GPL or the LGPL are applicable instead
00030  * of those above. If you wish to allow use of your version of this file only
00031  * under the terms of either the GPL or the LGPL, and not to allow others to
00032  * use your version of this file under the terms of the MPL, indicate your
00033  * decision by deleting the provisions above and replace them with the notice
00034  * and other provisions required by the GPL or the LGPL. If you do not delete
00035  * the provisions above, a recipient may use your version of this file under
00036  * the terms of any one of the MPL, the GPL or the LGPL.
00037  *
00038  * ***** END LICENSE BLOCK ***** */
00039 
00040 
00041 #include "nsTempfilePS.h"
00042 #include "nsCOMPtr.h"
00043 #include "nsDirectoryServiceDefs.h"
00044 #include "nsILocalFile.h"
00045 #include "nsPrintfCString.h"
00046 #include "prtime.h"
00047 
00048 
00049 nsTempfilePS::nsTempfilePS()
00050 {
00051     nsresult rv;
00052     
00053     // Get the standard temporary directory.
00054     rv = NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(mTempDir));
00055     if (NS_FAILED(rv))
00056         return;
00057 
00058     // Grab some low-order bits from the current time for use in temporary
00059     // filenames.
00060     LL_L2UI(mCount, PR_Now());
00061 
00062     // Append an arbitrary subdirectory name to the temp dir...
00063     rv = mTempDir->Append(
00064             NS_ConvertASCIItoUCS2(nsPrintfCString("%lx.tmp", mCount++)));
00065     if (NS_FAILED(rv)) {
00066         mTempDir = nsnull;
00067         return;
00068     }
00069     // and try to create it as a private directory for our temp files. Note
00070     // CreateUnique() may adjust the subdirectory name we just appended.
00071     rv = mTempDir->CreateUnique(nsIFile::DIRECTORY_TYPE, 0700);
00072     if (NS_FAILED(rv))
00073         mTempDir = nsnull;
00074 }
00075 
00076 nsTempfilePS::~nsTempfilePS()
00077 {
00078     if (nsnull != mTempDir)
00079         mTempDir->Remove(PR_TRUE);
00080 }
00081 
00082 nsresult
00083 nsTempfilePS::CreateTempFile(nsILocalFile** aFile)
00084 {
00085     NS_PRECONDITION(nsnull != aFile, "aFile argument is NULL");
00086     NS_ENSURE_TRUE(nsnull != mTempDir, NS_ERROR_NOT_INITIALIZED);
00087 
00088     // Get the temporary directory path
00089     nsresult rv;
00090     nsAutoString tmpdir;
00091     rv = mTempDir->GetPath(tmpdir);
00092     NS_ENSURE_SUCCESS(rv, rv);
00093 
00094     // Create a new object for the temporary file
00095     nsCOMPtr<nsILocalFile> tmpfile;
00096     rv = NS_NewLocalFile(tmpdir, PR_FALSE, getter_AddRefs(tmpfile));
00097     NS_ENSURE_SUCCESS(rv, rv);
00098     NS_POSTCONDITION(nsnull != tmpfile,
00099             "NS_NewLocalFile succeeded but tmpfile is invalid");
00100 
00101     rv = tmpfile->Append(
00102             NS_ConvertASCIItoUCS2(nsPrintfCString("%lx.tmp", mCount++)));
00103     NS_ENSURE_SUCCESS(rv, rv);
00104 
00105     // Create the temporary file. This may adjust the file's basename.
00106     rv = tmpfile->CreateUnique(nsIFile::NORMAL_FILE_TYPE, 0600);
00107     NS_ENSURE_SUCCESS(rv, rv);
00108 
00109     *aFile = tmpfile.get();
00110     NS_ADDREF(*aFile);
00111     return NS_OK;
00112 }
00113 
00114 
00115 nsresult
00116 nsTempfilePS::CreateTempFile(nsILocalFile** aFile,
00117         FILE **aHandle, const char *aMode)
00118 {
00119     NS_PRECONDITION(nsnull != aHandle, "aHandle is invalid");
00120     NS_PRECONDITION(nsnull != aMode, "aMode is invalid");
00121 
00122     nsresult rv = CreateTempFile(aFile);
00123     NS_ENSURE_SUCCESS(rv, rv);
00124     NS_POSTCONDITION(nsnull != *aFile,
00125             "CreateTempFile() succeeded but *aFile is invalid");
00126 
00127     rv = (*aFile)->OpenANSIFileDesc(aMode, aHandle);
00128     if (NS_FAILED(rv)) {
00129         (*aFile)->Remove(PR_FALSE);
00130         NS_RELEASE(*aFile);
00131     }
00132     return rv;
00133 }