mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
269 lines
9.5 KiB
C++
269 lines
9.5 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is Mozilla.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Mozilla Foundation.
|
|
* Portions created by the Initial Developer are Copyright (C) 2010 2011
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Wellington Fernando de Macedo <wfernandom2004@gmail.com> (original author)
|
|
* Patrick McManus <mcmanus@ducksong.com>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either of the GNU General Public License Version 2 or later (the "GPL"),
|
|
* or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
|
|
#include "nsIWebSocketProtocol.h"
|
|
#include "nsIURI.h"
|
|
#include "nsISupports.h"
|
|
#include "nsIInterfaceRequestor.h"
|
|
#include "nsIEventTarget.h"
|
|
#include "nsIStreamListener.h"
|
|
#include "nsIProtocolHandler.h"
|
|
#include "nsISocketTransport.h"
|
|
#include "nsIAsyncInputStream.h"
|
|
#include "nsIAsyncOutputStream.h"
|
|
#include "nsILoadGroup.h"
|
|
#include "nsITimer.h"
|
|
#include "nsIDNSListener.h"
|
|
#include "nsIHttpChannel.h"
|
|
#include "nsIChannelEventSink.h"
|
|
#include "nsIAsyncVerifyRedirectCallback.h"
|
|
#include "nsIStringStream.h"
|
|
#include "nsIHttpChannelInternal.h"
|
|
|
|
#include "nsCOMPtr.h"
|
|
#include "nsString.h"
|
|
#include "nsDeque.h"
|
|
|
|
namespace mozilla { namespace net {
|
|
|
|
class nsPostMessage;
|
|
class nsWSAdmissionManager;
|
|
class nsWSCompression;
|
|
class WSCallOnInputStreamReady;
|
|
|
|
class nsWebSocketHandler : public nsIWebSocketProtocol,
|
|
public nsIHttpUpgradeListener,
|
|
public nsIStreamListener,
|
|
public nsIProtocolHandler,
|
|
public nsIInputStreamCallback,
|
|
public nsIOutputStreamCallback,
|
|
public nsITimerCallback,
|
|
public nsIDNSListener,
|
|
public nsIInterfaceRequestor,
|
|
public nsIChannelEventSink
|
|
{
|
|
public:
|
|
NS_DECL_ISUPPORTS
|
|
NS_DECL_NSIWEBSOCKETPROTOCOL
|
|
NS_DECL_NSIHTTPUPGRADELISTENER
|
|
NS_DECL_NSIREQUESTOBSERVER
|
|
NS_DECL_NSISTREAMLISTENER
|
|
NS_DECL_NSIPROTOCOLHANDLER
|
|
NS_DECL_NSIINPUTSTREAMCALLBACK
|
|
NS_DECL_NSIOUTPUTSTREAMCALLBACK
|
|
NS_DECL_NSITIMERCALLBACK
|
|
NS_DECL_NSIDNSLISTENER
|
|
NS_DECL_NSIINTERFACEREQUESTOR
|
|
NS_DECL_NSICHANNELEVENTSINK
|
|
|
|
nsWebSocketHandler();
|
|
static void Shutdown();
|
|
|
|
enum {
|
|
// Non Control Frames
|
|
kContinuation = 0x0,
|
|
kText = 0x1,
|
|
kBinary = 0x2,
|
|
|
|
// Control Frames
|
|
kClose = 0x8,
|
|
kPing = 0x9,
|
|
kPong = 0xA
|
|
};
|
|
|
|
const static PRUint32 kControlFrameMask = 0x8;
|
|
const static PRInt32 kDefaultWSPort = 80;
|
|
const static PRInt32 kDefaultWSSPort = 443;
|
|
const static PRUint8 kMaskBit = 0x80;
|
|
const static PRUint8 kFinalFragBit = 0x80;
|
|
|
|
// section 7.4.1 defines these
|
|
const static PRUint16 kCloseNormal = 1000;
|
|
const static PRUint16 kCloseGoingAway = 1001;
|
|
const static PRUint16 kCloseProtocolError = 1002;
|
|
const static PRUint16 kCloseUnsupported = 1003;
|
|
const static PRUint16 kCloseTooLarge = 1004;
|
|
|
|
protected:
|
|
virtual ~nsWebSocketHandler();
|
|
PRBool mEncrypted;
|
|
|
|
private:
|
|
friend class nsPostMessage;
|
|
friend class nsWSAdmissionManager;
|
|
friend class WSCallOnInputStreamReady;
|
|
|
|
void SendMsgInternal(nsCString *aMsg, PRInt32 datalen);
|
|
void PrimeNewOutgoingMessage();
|
|
void GeneratePong(PRUint8 *payload, PRUint32 len);
|
|
void GeneratePing();
|
|
|
|
nsresult BeginOpen();
|
|
nsresult HandleExtensions();
|
|
nsresult SetupRequest();
|
|
nsresult ApplyForAdmission();
|
|
|
|
void StopSession(nsresult reason);
|
|
void AbortSession(nsresult reason);
|
|
void ReleaseSession();
|
|
|
|
void EnsureHdrOut(PRUint32 size);
|
|
void ApplyMask(PRUint32 mask, PRUint8 *data, PRUint64 len);
|
|
|
|
PRBool IsPersistentFramePtr();
|
|
nsresult ProcessInput(PRUint8 *buffer, PRUint32 count);
|
|
PRUint32 UpdateReadBuffer(PRUint8 *buffer, PRUint32 count);
|
|
|
|
class OutboundMessage
|
|
{
|
|
public:
|
|
OutboundMessage (nsCString *str)
|
|
: mMsg(str), mIsControl(PR_FALSE), mBinaryLen(-1) {}
|
|
|
|
OutboundMessage (nsCString *str, PRInt32 dataLen)
|
|
: mMsg(str), mIsControl(PR_FALSE), mBinaryLen(dataLen) {}
|
|
|
|
OutboundMessage ()
|
|
: mMsg(nsnull), mIsControl(PR_TRUE), mBinaryLen(-1) {}
|
|
|
|
~OutboundMessage() { delete mMsg; }
|
|
|
|
PRBool IsControl() { return mIsControl; }
|
|
const nsCString *Msg() { return mMsg; }
|
|
PRInt32 BinaryLen() { return mBinaryLen; }
|
|
PRInt32 Length()
|
|
{
|
|
if (mBinaryLen >= 0)
|
|
return mBinaryLen;
|
|
return mMsg ? mMsg->Length() : 0;
|
|
}
|
|
PRUint8 *BeginWriting()
|
|
{ return (PRUint8 *)(mMsg ? mMsg->BeginWriting() : nsnull); }
|
|
PRUint8 *BeginReading()
|
|
{ return (PRUint8 *)(mMsg ? mMsg->BeginReading() : nsnull); }
|
|
|
|
private:
|
|
nsCString *mMsg;
|
|
PRBool mIsControl;
|
|
PRInt32 mBinaryLen;
|
|
};
|
|
|
|
nsCOMPtr<nsIURI> mOriginalURI;
|
|
nsCOMPtr<nsIURI> mURI;
|
|
nsCOMPtr<nsIWebSocketListener> mListener;
|
|
nsCOMPtr<nsISupports> mContext;
|
|
nsCOMPtr<nsIInterfaceRequestor> mCallbacks;
|
|
nsCOMPtr<nsIEventTarget> mSocketThread;
|
|
nsCOMPtr<nsIHttpChannelInternal> mChannel;
|
|
nsCOMPtr<nsIHttpChannel> mHttpChannel;
|
|
nsCOMPtr<nsILoadGroup> mLoadGroup;
|
|
nsCOMPtr<nsICancelable> mDNSRequest;
|
|
nsCOMPtr<nsIAsyncVerifyRedirectCallback> mRedirectCallback;
|
|
|
|
nsCString mProtocol;
|
|
nsCString mOrigin;
|
|
nsCString mHashedSecret;
|
|
nsCString mAddress;
|
|
|
|
nsCOMPtr<nsISocketTransport> mTransport;
|
|
nsCOMPtr<nsIAsyncInputStream> mSocketIn;
|
|
nsCOMPtr<nsIAsyncOutputStream> mSocketOut;
|
|
|
|
nsCOMPtr<nsITimer> mCloseTimer;
|
|
PRUint32 mCloseTimeout; /* milliseconds */
|
|
|
|
nsCOMPtr<nsITimer> mOpenTimer;
|
|
PRUint32 mOpenTimeout; /* milliseconds */
|
|
|
|
nsCOMPtr<nsITimer> mPingTimer;
|
|
PRUint32 mPingTimeout; /* milliseconds */
|
|
PRUint32 mPingResponseTimeout; /* milliseconds */
|
|
|
|
PRUint32 mRecvdHttpOnStartRequest : 1;
|
|
PRUint32 mRecvdHttpUpgradeTransport : 1;
|
|
PRUint32 mRequestedClose : 1;
|
|
PRUint32 mClientClosed : 1;
|
|
PRUint32 mServerClosed : 1;
|
|
PRUint32 mStopped : 1;
|
|
PRUint32 mCalledOnStop : 1;
|
|
PRUint32 mPingOutstanding : 1;
|
|
PRUint32 mAllowCompression : 1;
|
|
PRUint32 mAutoFollowRedirects : 1;
|
|
PRUint32 mReleaseOnTransmit : 1;
|
|
|
|
PRInt32 mMaxMessageSize;
|
|
nsresult mStopOnClose;
|
|
|
|
// These are for the read buffers
|
|
PRUint8 *mFramePtr;
|
|
PRUint8 *mBuffer;
|
|
PRUint8 mFragmentOpcode;
|
|
PRUint32 mFragmentAccumulator;
|
|
PRUint32 mBuffered;
|
|
PRUint32 mBufferSize;
|
|
nsCOMPtr<nsIStreamListener> mInflateReader;
|
|
nsCOMPtr<nsIStringInputStream> mInflateStream;
|
|
|
|
// These are for the send buffers
|
|
const static PRInt32 kCopyBreak = 1000;
|
|
|
|
OutboundMessage *mCurrentOut;
|
|
PRUint32 mCurrentOutSent;
|
|
nsDeque mOutgoingMessages;
|
|
nsDeque mOutgoingPingMessages;
|
|
nsDeque mOutgoingPongMessages;
|
|
PRUint32 mHdrOutToSend;
|
|
PRUint8 *mHdrOut;
|
|
PRUint8 mOutHeader[kCopyBreak + 16];
|
|
nsWSCompression *mCompressor;
|
|
PRUint32 mDynamicOutputSize;
|
|
PRUint8 *mDynamicOutput;
|
|
};
|
|
|
|
class nsWebSocketSSLHandler : public nsWebSocketHandler
|
|
{
|
|
public:
|
|
nsWebSocketSSLHandler() {nsWebSocketHandler::mEncrypted = PR_TRUE;}
|
|
protected:
|
|
virtual ~nsWebSocketSSLHandler() {}
|
|
};
|
|
|
|
}} // namespace mozilla::net
|