The Tor Browser / file comparison
comparison: netwerk/base/src/nsDownloader.cpp
netwerk/base/src/nsDownloader.cpp
- branch
- TOR_BUG_9701
- changeset 15
- b8a032363ba2
equal
deleted
inserted
replaced
| -1:000000000000 | 0:511f130c35ed |
|---|---|
| 1 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 2 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 4 | |
| 5 #include "nsDownloader.h" | |
| 6 #include "nsIInputStream.h" | |
| 7 #include "nsDirectoryServiceUtils.h" | |
| 8 #include "nsDirectoryServiceDefs.h" | |
| 9 #include "nsNetUtil.h" | |
| 10 #include "nsCRTGlue.h" | |
| 11 | |
| 12 nsDownloader::~nsDownloader() | |
| 13 { | |
| 14 if (mLocation && mLocationIsTemp) { | |
| 15 // release the sink first since it may still hold an open file | |
| 16 // descriptor to mLocation. this needs to happen before the | |
| 17 // file can be removed otherwise the Remove call will fail. | |
| 18 if (mSink) { | |
| 19 mSink->Close(); | |
| 20 mSink = nullptr; | |
| 21 } | |
| 22 | |
| 23 nsresult rv = mLocation->Remove(false); | |
| 24 if (NS_FAILED(rv)) | |
| 25 NS_ERROR("unable to remove temp file"); | |
| 26 } | |
| 27 } | |
| 28 | |
| 29 NS_IMPL_ISUPPORTS(nsDownloader, | |
| 30 nsIDownloader, | |
| 31 nsIStreamListener, | |
| 32 nsIRequestObserver) | |
| 33 | |
| 34 NS_IMETHODIMP | |
| 35 nsDownloader::Init(nsIDownloadObserver *observer, nsIFile *location) | |
| 36 { | |
| 37 mObserver = observer; | |
| 38 mLocation = location; | |
| 39 return NS_OK; | |
| 40 } | |
| 41 | |
| 42 NS_IMETHODIMP | |
| 43 nsDownloader::OnStartRequest(nsIRequest *request, nsISupports *ctxt) | |
| 44 { | |
| 45 nsresult rv; | |
| 46 if (!mLocation) { | |
| 47 nsCOMPtr<nsIFile> location; | |
| 48 rv = NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(location)); | |
| 49 if (NS_FAILED(rv)) return rv; | |
| 50 | |
| 51 char buf[13]; | |
| 52 NS_MakeRandomString(buf, 8); | |
| 53 memcpy(buf+8, ".tmp", 5); | |
| 54 rv = location->AppendNative(nsDependentCString(buf, 12)); | |
| 55 if (NS_FAILED(rv)) return rv; | |
| 56 | |
| 57 rv = location->CreateUnique(nsIFile::NORMAL_FILE_TYPE, 0600); | |
| 58 if (NS_FAILED(rv)) return rv; | |
| 59 | |
| 60 location.swap(mLocation); | |
| 61 mLocationIsTemp = true; | |
| 62 } | |
| 63 | |
| 64 rv = NS_NewLocalFileOutputStream(getter_AddRefs(mSink), mLocation); | |
| 65 if (NS_FAILED(rv)) return rv; | |
| 66 | |
| 67 // we could wrap this output stream with a buffered output stream, | |
| 68 // but it shouldn't be necessary since we will be writing large | |
| 69 // chunks given to us via OnDataAvailable. | |
| 70 | |
| 71 return NS_OK; | |
| 72 } | |
| 73 | |
| 74 NS_IMETHODIMP | |
| 75 nsDownloader::OnStopRequest(nsIRequest *request, | |
| 76 nsISupports *ctxt, | |
| 77 nsresult status) | |
| 78 { | |
| 79 if (mSink) { | |
| 80 mSink->Close(); | |
| 81 mSink = nullptr; | |
| 82 } | |
| 83 | |
| 84 mObserver->OnDownloadComplete(this, request, ctxt, status, mLocation); | |
| 85 mObserver = nullptr; | |
| 86 | |
| 87 return NS_OK; | |
| 88 } | |
| 89 | |
| 90 NS_METHOD | |
| 91 nsDownloader::ConsumeData(nsIInputStream* in, | |
| 92 void* closure, | |
| 93 const char* fromRawSegment, | |
| 94 uint32_t toOffset, | |
| 95 uint32_t count, | |
| 96 uint32_t *writeCount) | |
| 97 { | |
| 98 nsDownloader *self = (nsDownloader *) closure; | |
| 99 if (self->mSink) | |
| 100 return self->mSink->Write(fromRawSegment, count, writeCount); | |
| 101 | |
| 102 *writeCount = count; | |
| 103 return NS_OK; | |
| 104 } | |
| 105 | |
| 106 NS_IMETHODIMP | |
| 107 nsDownloader::OnDataAvailable(nsIRequest *request, nsISupports *ctxt, | |
| 108 nsIInputStream *inStr, | |
| 109 uint64_t sourceOffset, uint32_t count) | |
| 110 { | |
| 111 uint32_t n; | |
| 112 return inStr->ReadSegments(ConsumeData, this, count, &n); | |
| 113 } |
