1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/toolkit/components/url-classifier/nsCheckSummedOutputStream.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,59 @@
1.4 +//* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "nsILocalFile.h"
1.10 +#include "nsCRT.h"
1.11 +#include "nsIFile.h"
1.12 +#include "nsISupportsImpl.h"
1.13 +#include "nsCheckSummedOutputStream.h"
1.14 +
1.15 +////////////////////////////////////////////////////////////////////////////////
1.16 +// nsCheckSummedOutputStream
1.17 +
1.18 +NS_IMPL_ISUPPORTS_INHERITED(nsCheckSummedOutputStream,
1.19 + nsSafeFileOutputStream,
1.20 + nsISafeOutputStream,
1.21 + nsIOutputStream,
1.22 + nsIFileOutputStream)
1.23 +
1.24 +NS_IMETHODIMP
1.25 +nsCheckSummedOutputStream::Init(nsIFile* file, int32_t ioFlags, int32_t perm,
1.26 + int32_t behaviorFlags)
1.27 +{
1.28 + nsresult rv;
1.29 + mHash = do_CreateInstance(NS_CRYPTO_HASH_CONTRACTID, &rv);
1.30 + NS_ENSURE_SUCCESS(rv, rv);
1.31 +
1.32 + rv = mHash->Init(nsICryptoHash::MD5);
1.33 + NS_ENSURE_SUCCESS(rv, rv);
1.34 +
1.35 + return nsSafeFileOutputStream::Init(file, ioFlags, perm, behaviorFlags);
1.36 +}
1.37 +
1.38 +NS_IMETHODIMP
1.39 +nsCheckSummedOutputStream::Finish()
1.40 +{
1.41 + nsresult rv = mHash->Finish(false, mCheckSum);
1.42 + NS_ENSURE_SUCCESS(rv, rv);
1.43 +
1.44 + uint32_t written;
1.45 + rv = nsSafeFileOutputStream::Write(reinterpret_cast<const char*>(mCheckSum.BeginReading()),
1.46 + mCheckSum.Length(), &written);
1.47 + NS_ASSERTION(written == mCheckSum.Length(), "Error writing stream checksum");
1.48 + NS_ENSURE_SUCCESS(rv, rv);
1.49 +
1.50 + return nsSafeFileOutputStream::Finish();
1.51 +}
1.52 +
1.53 +NS_IMETHODIMP
1.54 +nsCheckSummedOutputStream::Write(const char *buf, uint32_t count, uint32_t *result)
1.55 +{
1.56 + nsresult rv = mHash->Update(reinterpret_cast<const uint8_t*>(buf), count);
1.57 + NS_ENSURE_SUCCESS(rv, rv);
1.58 +
1.59 + return nsSafeFileOutputStream::Write(buf, count, result);
1.60 +}
1.61 +
1.62 +////////////////////////////////////////////////////////////////////////////////
