/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /* ***** BEGIN LICENSE BLOCK ***** * Version: NPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Netscape Public License * Version 1.1 (the "License"); you may not use this file except in * compliance with the License. You may obtain a copy of the License at * http://www.mozilla.org/NPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the * License. * * The Original Code is mozilla.org code. * * The Initial Developer of the Original Code is * Netscape Communications Corporation. * Portions created by the Initial Developer are Copyright (C) 1998 * the Initial Developer. All Rights Reserved. * * Contributor(s): * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the NPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the NPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #include "nsIServiceManager.h" #include "nsIComponentRegistrar.h" #include "nsIInputStream.h" #include "nsIOutputStream.h" #include "nsIRunnable.h" #include "nsIThread.h" #include "nsISupportsArray.h" #include "prinrval.h" #include "nsIFileStreams.h" #include "nsIFileChannel.h" #include "nsILocalFile.h" #include "nsNetUtil.h" #include //////////////////////////////////////////////////////////////////////////////// #include #include "prprf.h" #include "nsAutoLock.h" class nsTimeSampler { public: nsTimeSampler(); void Reset(); void StartTime(); void EndTime(); void AddTime(PRIntervalTime time); PRIntervalTime LastInterval() { return mLastInterval; } char* PrintStats(); protected: PRIntervalTime mStartTime; double mSquares; double mTotalTime; PRUint32 mCount; PRIntervalTime mLastInterval; }; nsTimeSampler::nsTimeSampler() { Reset(); } void nsTimeSampler::Reset() { mStartTime = 0; mSquares = 0; mTotalTime = 0; mCount = 0; mLastInterval = 0; } void nsTimeSampler::StartTime() { mStartTime = PR_IntervalNow(); } void nsTimeSampler::EndTime() { NS_ASSERTION(mStartTime != 0, "Forgot to call StartTime"); PRIntervalTime endTime = PR_IntervalNow(); mLastInterval = endTime - mStartTime; AddTime(mLastInterval); mStartTime = 0; } void nsTimeSampler::AddTime(PRIntervalTime time) { nsAutoCMonitor mon(this); mTotalTime += time; mSquares += (double)time * (double)time; mCount++; } char* nsTimeSampler::PrintStats() { double mean = mTotalTime / mCount; double variance = fabs(mSquares / mCount - mean * mean); double stddev = sqrt(variance); PRUint32 imean = (PRUint32)mean; PRUint32 istddev = (PRUint32)stddev; return PR_smprintf("%d +/- %d ms", PR_IntervalToMilliseconds(imean), PR_IntervalToMilliseconds(istddev)); } //////////////////////////////////////////////////////////////////////////////// nsTimeSampler gTimeSampler; typedef nsresult (*CreateFun)(nsIRunnable* *result, nsIFile* inPath, nsIFile* outPath, PRUint32 bufferSize); //////////////////////////////////////////////////////////////////////////////// nsresult Copy(nsIInputStream* inStr, nsIOutputStream* outStr, char* buf, PRUint32 bufSize, PRUint32 *copyCount) { nsresult rv; while (PR_TRUE) { PRUint32 count; rv = inStr->Read(buf, bufSize, &count); if (NS_FAILED(rv)) return rv; if (count == 0) break; PRUint32 writeCount; rv = outStr->Write(buf, count, &writeCount); if (NS_FAILED(rv)) return rv; NS_ASSERTION(writeCount == count, "didn't write all the data"); *copyCount += writeCount; } rv = outStr->Flush(); return rv; } //////////////////////////////////////////////////////////////////////////////// class FileSpecWorker : public nsIRunnable { public: NS_IMETHOD Run() { nsresult rv; PRIntervalTime startTime = PR_IntervalNow(); PRIntervalTime endTime; nsCOMPtr inStr; nsCOMPtr outStr; PRUint32 copyCount = 0; // Open the input stream: nsCOMPtr fileIn; rv = NS_NewLocalFileInputStream(getter_AddRefs(fileIn), mInPath); if (NS_FAILED(rv)) return rv; rv = NS_NewBufferedInputStream(getter_AddRefs(inStr), fileIn, 65535); if (NS_FAILED(rv)) return rv; // Open the output stream: nsCOMPtr fileOut; rv = NS_NewLocalFileOutputStream(getter_AddRefs(fileOut), mOutPath, PR_CREATE_FILE | PR_WRONLY | PR_TRUNCATE, 0664); if (NS_FAILED(rv)) return rv; rv = NS_NewBufferedOutputStream(getter_AddRefs(outStr), fileOut, 65535); if (NS_FAILED(rv)) return rv; // Copy from one to the other rv = Copy(inStr, outStr, mBuffer, mBufferSize, ©Count); if (NS_FAILED(rv)) return rv; endTime = PR_IntervalNow(); gTimeSampler.AddTime(endTime - startTime); return rv; } NS_DECL_ISUPPORTS FileSpecWorker() : mInPath(nsnull), mOutPath(nsnull), mBuffer(nsnull), mBufferSize(0) { NS_INIT_REFCNT(); } nsresult Init(nsIFile* inPath, nsIFile* outPath, PRUint32 bufferSize) { mInPath = inPath; mOutPath = outPath; mBuffer = new char[bufferSize]; mBufferSize = bufferSize; return (mInPath && mOutPath && mBuffer) ? NS_OK : NS_ERROR_OUT_OF_MEMORY; } static nsresult Create(nsIRunnable* *result, nsIFile* inPath, nsIFile* outPath, PRUint32 bufferSize) { FileSpecWorker* worker = new FileSpecWorker(); if (worker == nsnull) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(worker); nsresult rv = worker->Init(inPath, outPath, bufferSize); if (NS_FAILED(rv)) { NS_RELEASE(worker); return rv; } *result = worker; return NS_OK; } virtual ~FileSpecWorker() { delete[] mBuffer; } protected: nsCOMPtr mInPath; nsCOMPtr mOutPath; char* mBuffer; PRUint32 mBufferSize; }; NS_IMPL_ISUPPORTS1(FileSpecWorker, nsIRunnable) //////////////////////////////////////////////////////////////////////////////// #include "nsIIOService.h" #include "nsIChannel.h" class FileChannelWorker : public nsIRunnable { public: NS_IMETHOD Run() { nsresult rv; PRIntervalTime startTime = PR_IntervalNow(); PRIntervalTime endTime; PRUint32 copyCount = 0; nsCOMPtr inCh; nsCOMPtr outCh; nsCOMPtr inStr; nsCOMPtr outStr; rv = NS_NewLocalFileChannel(getter_AddRefs(inCh), mInPath); if (NS_FAILED(rv)) return rv; rv = inCh->Open(getter_AddRefs(inStr)); if (NS_FAILED(rv)) return rv; //rv = NS_NewLocalFileChannel(getter_AddRefs(outCh), mOutPath); //if (NS_FAILED(rv)) return rv; //rv = outCh->OpenOutputStream(0, -1, 0, getter_AddRefs(outStr)); //if (NS_FAILED(rv)) return rv; // Copy from one to the other rv = Copy(inStr, outStr, mBuffer, mBufferSize, ©Count); if (NS_FAILED(rv)) return rv; endTime = PR_IntervalNow(); gTimeSampler.AddTime(endTime - startTime); return rv; } NS_DECL_ISUPPORTS FileChannelWorker() : mInPath(nsnull), mOutPath(nsnull), mBuffer(nsnull), mBufferSize(0) { NS_INIT_REFCNT(); } nsresult Init(nsIFile* inPath, nsIFile* outPath, PRUint32 bufferSize) { mInPath = inPath; mOutPath = outPath; mBuffer = new char[bufferSize]; mBufferSize = bufferSize; return (mInPath && mOutPath && mBuffer) ? NS_OK : NS_ERROR_OUT_OF_MEMORY; } static nsresult Create(nsIRunnable* *result, nsIFile* inPath, nsIFile* outPath, PRUint32 bufferSize) { FileChannelWorker* worker = new FileChannelWorker(); if (worker == nsnull) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(worker); nsresult rv = worker->Init(inPath, outPath, bufferSize); if (NS_FAILED(rv)) { NS_RELEASE(worker); return rv; } *result = worker; return NS_OK; } virtual ~FileChannelWorker() { delete[] mBuffer; } protected: nsCOMPtr mInPath; nsCOMPtr mOutPath; char* mBuffer; PRUint32 mBufferSize; }; NS_IMPL_ISUPPORTS1(FileChannelWorker, nsIRunnable) //////////////////////////////////////////////////////////////////////////////// void Test(CreateFun create, PRUint32 count, nsIFile* inDirSpec, nsIFile* outDirSpec, PRUint32 bufSize) { nsresult rv; PRUint32 i; nsCAutoString inDir; nsCAutoString outDir; (void)inDirSpec->GetNativePath(inDir); (void)outDirSpec->GetNativePath(outDir); printf("###########\nTest: from %s to %s, bufSize = %d\n", inDir.get(), outDir.get(), bufSize); gTimeSampler.Reset(); nsTimeSampler testTime; testTime.StartTime(); nsISupportsArray* threads; rv = NS_NewISupportsArray(&threads); NS_ASSERTION(NS_SUCCEEDED(rv), "NS_NewISupportsArray failed"); nsCOMPtr entries; rv = inDirSpec->GetDirectoryEntries(getter_AddRefs(entries)); NS_ASSERTION(NS_SUCCEEDED(rv), "GetDirectoryEntries failed"); i = 0; PRBool hasMore; while (i < count && NS_SUCCEEDED(entries->HasMoreElements(&hasMore)) && hasMore) { nsCOMPtr next; rv = entries->GetNext(getter_AddRefs(next)); if (NS_FAILED(rv)) goto done; nsCOMPtr inSpec = do_QueryInterface(next, &rv); if (NS_FAILED(rv)) goto done; nsCOMPtr outSpec; rv = outDirSpec->Clone(getter_AddRefs(outSpec)); // don't munge the original if (NS_FAILED(rv)) goto done; nsCAutoString leafName; rv = inSpec->GetNativeLeafName(leafName); if (NS_FAILED(rv)) goto done; rv = outSpec->AppendNative(leafName); if (NS_FAILED(rv)) goto done; PRBool exists; rv = outSpec->Exists(&exists); if (NS_FAILED(rv)) goto done; if (exists) { rv = outSpec->Remove(PR_FALSE); if (NS_FAILED(rv)) goto done; } nsCOMPtr thread; nsCOMPtr worker; rv = create(getter_AddRefs(worker), inSpec, outSpec, bufSize); if (NS_FAILED(rv)) goto done; rv = NS_NewThread(getter_AddRefs(thread), worker, 0, PR_JOINABLE_THREAD); if (NS_FAILED(rv)) goto done; PRBool inserted = threads->InsertElementAt(thread, i); NS_ASSERTION(inserted, "not inserted"); i++; } PRUint32 j; for (j = 0; j < i; j++) { nsIThread* thread = (nsIThread*)threads->ElementAt(j); if (NS_FAILED(rv)) goto done; thread->Join(); } done: NS_RELEASE(threads); NS_ASSERTION(rv == NS_OK, "failed"); testTime.EndTime(); char* testStats = testTime.PrintStats(); char* workerStats = gTimeSampler.PrintStats(); printf(" threads = %d\n work time = %s,\n test time = %s\n", i, workerStats, testStats); PR_smprintf_free(workerStats); PR_smprintf_free(testStats); } //////////////////////////////////////////////////////////////////////////////// int main(int argc, char* argv[]) { nsresult rv; if (argc < 2) { printf("usage: %s \n", argv[0]); return -1; } char* inDir = argv[1]; char* outDir = argv[2]; { nsCOMPtr servMan; NS_InitXPCOM2(getter_AddRefs(servMan), nsnull, nsnull); nsCOMPtr registrar = do_QueryInterface(servMan); NS_ASSERTION(registrar, "Null nsIComponentRegistrar"); if (registrar) registrar->AutoRegister(nsnull); nsCOMPtr inDirFile; rv = NS_NewNativeLocalFile(nsDependentCString(inDir), PR_FALSE, getter_AddRefs(inDirFile)); if (NS_FAILED(rv)) return rv; nsCOMPtr outDirFile; rv = NS_NewNativeLocalFile(nsDependentCString(outDir), PR_FALSE, getter_AddRefs(outDirFile)); if (NS_FAILED(rv)) return rv; CreateFun create = FileChannelWorker::Create; Test(create, 1, inDirFile, outDirFile, 16 * 1024); #if 1 printf("FileChannelWorker *****************************\n"); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); #endif create = FileSpecWorker::Create; printf("FileSpecWorker ********************************\n"); #if 1 Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); Test(create, 20, inDirFile, outDirFile, 16 * 1024); #endif #if 1 Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); Test(create, 20, inDirFile, outDirFile, 4 * 1024); #endif } // this scopes the nsCOMPtrs // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM rv = NS_ShutdownXPCOM(nsnull); NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed"); return 0; } ////////////////////////////////////////////////////////////////////////////////