Line data    Source code

       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 "nsToolkitCompsCID.h"
      16             : #include "../../../netwerk/base/nsBufferedStreams.h"
      17             : #include "prio.h"
      18             : 
      19             : class nsCheckSummedOutputStream : public nsBufferedOutputStream
      20             : {
      21             : public:
      22             :   NS_DECL_ISUPPORTS_INHERITED
      23             : 
      24             :   // Size of MD5 hash in bytes
      25             :   static const uint32_t CHECKSUM_SIZE = 16;
      26             :   static const uint32_t MAX_BUFFER_SIZE = 64 * 1024;
      27             : 
      28           6 :   nsCheckSummedOutputStream() {}
      29             : 
      30             :   NS_IMETHOD Finish() override;
      31             :   NS_IMETHOD Write(const char *buf, uint32_t count, uint32_t *result) override;
      32             :   NS_IMETHOD Init(nsIOutputStream* stream, uint32_t bufferSize) override;
      33             : 
      34             : protected:
      35          18 :   virtual ~nsCheckSummedOutputStream() { nsBufferedOutputStream::Close(); }
      36             : 
      37             :   nsCOMPtr<nsICryptoHash> mHash;
      38             :   nsCString mCheckSum;
      39             : };
      40             : 
      41             : // returns a file output stream which can be QI'ed to nsIFileOutputStream.
      42             : inline nsresult
      43           6 : NS_NewCheckSummedOutputStream(nsIOutputStream **result,
      44             :                               nsIFile         *file)
      45             : {
      46          12 :     nsCOMPtr<nsIOutputStream> localOutFile;
      47          12 :     nsresult rv = NS_NewSafeLocalFileOutputStream(getter_AddRefs(localOutFile), file,
      48           6 :                                                   PR_WRONLY | PR_TRUNCATE | PR_CREATE_FILE);
      49           6 :     NS_ENSURE_SUCCESS(rv, rv);
      50             : 
      51          12 :     nsCOMPtr<nsIBufferedOutputStream> out = new nsCheckSummedOutputStream();
      52           6 :     rv = out->Init(localOutFile, nsCheckSummedOutputStream::CHECKSUM_SIZE);
      53           6 :     if (NS_SUCCEEDED(rv)) {
      54           6 :       out.forget(result);
      55             :     }
      56           6 :     return rv;
      57             : }
      58             : 
      59             : #endif