forked from mirrors/gecko-dev
c1fae83952
Backed out changeset fce62c77a56b (bug 1478124) Backed out changeset eb2fa3b5edf7 (bug 1478124) Backed out changeset 8dacce59fcc0 (bug 1478124) Backed out changeset 012fd0107204 (bug 1478124) Backed out changeset 496aaf774697 (bug 1478124) Backed out changeset 21f4fda03159 (bug 1478124) Backed out changeset b0444e0bc801 (bug 1478124) Backed out changeset d94039b19943 (bug 1478124) Backed out changeset 5d85deac61c2 (bug 1478124) Backed out changeset 929fd654c9df (bug 1478124) Backed out changeset 1ddd80d9e91a (bug 1478124) Backed out changeset b8d2dfdfc324 (bug 1478124) Backed out changeset f500020a273a (bug 1478124) Backed out changeset dd00365ebb55 (bug 1478124) Backed out changeset 538e40c5ee13 (bug 1478124) Backed out changeset bedaa9c437ad (bug 1478124)
56 lines
1.8 KiB
C++
56 lines
1.8 KiB
C++
//* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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/. */
|
|
|
|
#ifndef nsCheckSummedOutputStream_h__
|
|
#define nsCheckSummedOutputStream_h__
|
|
|
|
#include "nsIFile.h"
|
|
#include "nsIOutputStream.h"
|
|
#include "nsICryptoHash.h"
|
|
#include "nsNetCID.h"
|
|
#include "nsNetUtil.h"
|
|
#include "nsString.h"
|
|
#include "../../../netwerk/base/nsBufferedStreams.h"
|
|
#include "prio.h"
|
|
|
|
class nsCheckSummedOutputStream : public nsBufferedOutputStream {
|
|
public:
|
|
NS_DECL_ISUPPORTS_INHERITED
|
|
|
|
// Size of MD5 hash in bytes
|
|
static const uint32_t CHECKSUM_SIZE = 16;
|
|
static const uint32_t MAX_BUFFER_SIZE = 64 * 1024;
|
|
|
|
nsCheckSummedOutputStream() {}
|
|
|
|
NS_IMETHOD Finish() override;
|
|
NS_IMETHOD Write(const char *buf, uint32_t count, uint32_t *result) override;
|
|
NS_IMETHOD Init(nsIOutputStream *stream, uint32_t bufferSize) override;
|
|
|
|
protected:
|
|
virtual ~nsCheckSummedOutputStream() { nsBufferedOutputStream::Close(); }
|
|
|
|
nsCOMPtr<nsICryptoHash> mHash;
|
|
nsCString mCheckSum;
|
|
};
|
|
|
|
// returns a file output stream which can be QI'ed to nsIFileOutputStream.
|
|
inline nsresult NS_NewCheckSummedOutputStream(nsIOutputStream **result,
|
|
nsIFile *file) {
|
|
nsCOMPtr<nsIOutputStream> localOutFile;
|
|
nsresult rv =
|
|
NS_NewSafeLocalFileOutputStream(getter_AddRefs(localOutFile), file,
|
|
PR_WRONLY | PR_TRUNCATE | PR_CREATE_FILE);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIBufferedOutputStream> out = new nsCheckSummedOutputStream();
|
|
rv = out->Init(localOutFile, nsCheckSummedOutputStream::CHECKSUM_SIZE);
|
|
if (NS_SUCCEEDED(rv)) {
|
|
out.forget(result);
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
#endif
|