gecko-dev/netwerk/base/nsDownloader.cpp
Gabriele Svelto 5dc21d568c Bug 1600545 - Remove useless inclusions of header files generated from IDL files in modules/, netwerk/, parser/, security/, startupcache/, storage/, toolkit/, tools/, uriloader/, widget/, xpcom/ and xpfe/ r=Ehsan
The inclusions were removed with the following very crude script and the
resulting breakage was fixed up by hand. The manual fixups did either
revert the changes done by the script, replace a generic header with a more
specific one or replace a header with a forward declaration.

find . -name "*.idl" | grep -v web-platform | grep -v third_party | while read path; do
    interfaces=$(grep "^\(class\|interface\).*:.*" "$path" | cut -d' ' -f2)
    if [ -n "$interfaces" ]; then
        if [[ "$interfaces" == *$'\n'* ]]; then
          regexp="\("
          for i in $interfaces; do regexp="$regexp$i\|"; done
          regexp="${regexp%%\\\|}\)"
        else
          regexp="$interfaces"
        fi
        interface=$(basename "$path")
        rg -l "#include.*${interface%%.idl}.h" . | while read path2; do
            hits=$(grep -v "#include.*${interface%%.idl}.h" "$path2" | grep -c "$regexp" )
            if [ $hits -eq 0 ]; then
                echo "Removing ${interface} from ${path2}"
                grep -v "#include.*${interface%%.idl}.h" "$path2" > "$path2".tmp
                mv -f "$path2".tmp "$path2"
            fi
        done
    fi
done

Differential Revision: https://phabricator.services.mozilla.com/D55444

--HG--
extra : moz-landing-system : lando
2019-12-06 09:17:57 +00:00

98 lines
2.8 KiB
C++

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "nsDownloader.h"
#include "nsIInputStream.h"
#include "nsDirectoryServiceUtils.h"
#include "nsDirectoryServiceDefs.h"
#include "nsNetUtil.h"
#include "nsCRTGlue.h"
nsDownloader::~nsDownloader() {
if (mLocation && mLocationIsTemp) {
// release the sink first since it may still hold an open file
// descriptor to mLocation. this needs to happen before the
// file can be removed otherwise the Remove call will fail.
if (mSink) {
mSink->Close();
mSink = nullptr;
}
nsresult rv = mLocation->Remove(false);
if (NS_FAILED(rv)) NS_ERROR("unable to remove temp file");
}
}
NS_IMPL_ISUPPORTS(nsDownloader, nsIDownloader, nsIStreamListener,
nsIRequestObserver)
NS_IMETHODIMP
nsDownloader::Init(nsIDownloadObserver* observer, nsIFile* location) {
mObserver = observer;
mLocation = location;
return NS_OK;
}
NS_IMETHODIMP
nsDownloader::OnStartRequest(nsIRequest* request) {
nsresult rv;
if (!mLocation) {
nsCOMPtr<nsIFile> location;
rv = NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(location));
if (NS_FAILED(rv)) return rv;
char buf[13];
NS_MakeRandomString(buf, 8);
memcpy(buf + 8, ".tmp", 5);
rv = location->AppendNative(nsDependentCString(buf, 12));
if (NS_FAILED(rv)) return rv;
rv = location->CreateUnique(nsIFile::NORMAL_FILE_TYPE, 0600);
if (NS_FAILED(rv)) return rv;
location.swap(mLocation);
mLocationIsTemp = true;
}
rv = NS_NewLocalFileOutputStream(getter_AddRefs(mSink), mLocation);
if (NS_FAILED(rv)) return rv;
// we could wrap this output stream with a buffered output stream,
// but it shouldn't be necessary since we will be writing large
// chunks given to us via OnDataAvailable.
return NS_OK;
}
NS_IMETHODIMP
nsDownloader::OnStopRequest(nsIRequest* request, nsresult status) {
if (mSink) {
mSink->Close();
mSink = nullptr;
}
mObserver->OnDownloadComplete(this, request, nullptr, status, mLocation);
mObserver = nullptr;
return NS_OK;
}
nsresult nsDownloader::ConsumeData(nsIInputStream* in, void* closure,
const char* fromRawSegment,
uint32_t toOffset, uint32_t count,
uint32_t* writeCount) {
nsDownloader* self = (nsDownloader*)closure;
if (self->mSink) return self->mSink->Write(fromRawSegment, count, writeCount);
*writeCount = count;
return NS_OK;
}
NS_IMETHODIMP
nsDownloader::OnDataAvailable(nsIRequest* request, nsIInputStream* inStr,
uint64_t sourceOffset, uint32_t count) {
uint32_t n;
return inStr->ReadSegments(ConsumeData, this, count, &n);
}