diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /netwerk/streamconv/converters/nsHTTPCompressConv.h | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | uxp-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz |
Add m-esr52 at 52.6.0
Diffstat (limited to 'netwerk/streamconv/converters/nsHTTPCompressConv.h')
-rw-r--r-- | netwerk/streamconv/converters/nsHTTPCompressConv.h | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/netwerk/streamconv/converters/nsHTTPCompressConv.h b/netwerk/streamconv/converters/nsHTTPCompressConv.h new file mode 100644 index 0000000000..38889a8f71 --- /dev/null +++ b/netwerk/streamconv/converters/nsHTTPCompressConv.h @@ -0,0 +1,135 @@ +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set sw=2 ts=8 et tw=80 : */ +/* 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/. */ + +#if !defined (__nsHTTPCompressConv__h__) +#define __nsHTTPCompressConv__h__ 1 + +#include "nsIStreamConverter.h" +#include "nsICompressConvStats.h" +#include "nsCOMPtr.h" +#include "nsAutoPtr.h" + +#include "zlib.h" + +// brotli includes +#undef assert +#include "assert.h" +#include "state.h" + +class nsIStringInputStream; + +#define NS_HTTPCOMPRESSCONVERTER_CID \ + { \ + /* 66230b2b-17fa-4bd3-abf4-07986151022d */ \ + 0x66230b2b, \ + 0x17fa, \ + 0x4bd3, \ + {0xab, 0xf4, 0x07, 0x98, 0x61, 0x51, 0x02, 0x2d} \ + } + + +#define HTTP_DEFLATE_TYPE "deflate" +#define HTTP_GZIP_TYPE "gzip" +#define HTTP_X_GZIP_TYPE "x-gzip" +#define HTTP_COMPRESS_TYPE "compress" +#define HTTP_X_COMPRESS_TYPE "x-compress" +#define HTTP_BROTLI_TYPE "br" +#define HTTP_IDENTITY_TYPE "identity" +#define HTTP_UNCOMPRESSED_TYPE "uncompressed" + +namespace mozilla { +namespace net { + +typedef enum { + HTTP_COMPRESS_GZIP, + HTTP_COMPRESS_DEFLATE, + HTTP_COMPRESS_COMPRESS, + HTTP_COMPRESS_BROTLI, + HTTP_COMPRESS_IDENTITY +} CompressMode; + +class BrotliWrapper +{ +public: + BrotliWrapper() + : mTotalOut(0) + , mStatus(NS_OK) + { + BrotliStateInit(&mState); + } + ~BrotliWrapper() + { + BrotliStateCleanup(&mState); + } + + BrotliState mState; + size_t mTotalOut; + nsresult mStatus; + + nsIRequest *mRequest; + nsISupports *mContext; + uint64_t mSourceOffset; +}; + +class nsHTTPCompressConv + : public nsIStreamConverter + , public nsICompressConvStats +{ + public: + // nsISupports methods + NS_DECL_THREADSAFE_ISUPPORTS + NS_DECL_NSIREQUESTOBSERVER + NS_DECL_NSISTREAMLISTENER + NS_DECL_NSICOMPRESSCONVSTATS + + // nsIStreamConverter methods + NS_DECL_NSISTREAMCONVERTER + + nsHTTPCompressConv (); + +private: + virtual ~nsHTTPCompressConv (); + + nsCOMPtr<nsIStreamListener> mListener; // this guy gets the converted data via his OnDataAvailable () + CompressMode mMode; + + unsigned char *mOutBuffer; + unsigned char *mInpBuffer; + + uint32_t mOutBufferLen; + uint32_t mInpBufferLen; + + nsAutoPtr<BrotliWrapper> mBrotli; + + nsCOMPtr<nsISupports> mAsyncConvContext; + nsCOMPtr<nsIStringInputStream> mStream; + + static nsresult + BrotliHandler(nsIInputStream *stream, void *closure, const char *dataIn, + uint32_t, uint32_t avail, uint32_t *countRead); + + nsresult do_OnDataAvailable (nsIRequest *request, nsISupports *aContext, + uint64_t aSourceOffset, const char *buffer, + uint32_t aCount); + + bool mCheckHeaderDone; + bool mStreamEnded; + bool mStreamInitialized; + bool mDummyStreamInitialised; + bool mFailUncleanStops; + + z_stream d_stream; + unsigned mLen, hMode, mSkipCount, mFlags; + + uint32_t check_header (nsIInputStream *iStr, uint32_t streamLen, nsresult *rv); + + uint32_t mDecodedDataLength; +}; + +} // namespace net +} // namespace mozilla + +#endif |