The Tor Browser / file comparison
comparison: toolkit/components/url-classifier/nsCheckSummedOutputStream.h
toolkit/components/url-classifier/nsCheckSummedOutputStream.h
- branch
- TOR_BUG_9701
- changeset 14
- 925c144e1f1f
equal
deleted
inserted
replaced
| -1:000000000000 | 0:b700f1499a51 |
|---|---|
| 1 //* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ | |
| 2 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 3 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 5 | |
| 6 #ifndef nsCheckSummedOutputStream_h__ | |
| 7 #define nsCheckSummedOutputStream_h__ | |
| 8 | |
| 9 #include "nsILocalFile.h" | |
| 10 #include "nsIFile.h" | |
| 11 #include "nsIOutputStream.h" | |
| 12 #include "nsICryptoHash.h" | |
| 13 #include "nsNetCID.h" | |
| 14 #include "nsString.h" | |
| 15 #include "../../../netwerk/base/src/nsFileStreams.h" | |
| 16 #include "nsToolkitCompsCID.h" | |
| 17 | |
| 18 class nsCheckSummedOutputStream : public nsSafeFileOutputStream | |
| 19 { | |
| 20 public: | |
| 21 NS_DECL_ISUPPORTS_INHERITED | |
| 22 | |
| 23 // Size of MD5 hash in bytes | |
| 24 static const uint32_t CHECKSUM_SIZE = 16; | |
| 25 | |
| 26 nsCheckSummedOutputStream() {} | |
| 27 virtual ~nsCheckSummedOutputStream() { nsSafeFileOutputStream::Close(); } | |
| 28 | |
| 29 NS_IMETHOD Finish(); | |
| 30 NS_IMETHOD Write(const char *buf, uint32_t count, uint32_t *result); | |
| 31 NS_IMETHOD Init(nsIFile* file, int32_t ioFlags, int32_t perm, int32_t behaviorFlags); | |
| 32 | |
| 33 protected: | |
| 34 nsCOMPtr<nsICryptoHash> mHash; | |
| 35 nsAutoCString mCheckSum; | |
| 36 }; | |
| 37 | |
| 38 // returns a file output stream which can be QI'ed to nsIFileOutputStream. | |
| 39 inline nsresult | |
| 40 NS_NewCheckSummedOutputStream(nsIOutputStream **result, | |
| 41 nsIFile *file, | |
| 42 int32_t ioFlags = -1, | |
| 43 int32_t perm = -1, | |
| 44 int32_t behaviorFlags = 0) | |
| 45 { | |
| 46 nsCOMPtr<nsIFileOutputStream> out = new nsCheckSummedOutputStream(); | |
| 47 nsresult rv = out->Init(file, ioFlags, perm, behaviorFlags); | |
| 48 if (NS_SUCCEEDED(rv)) | |
| 49 NS_ADDREF(*result = out); // cannot use nsCOMPtr::swap | |
| 50 return rv; | |
| 51 } | |
| 52 | |
| 53 #endif |
