зеркало из https://github.com/mozilla/gecko-dev.git
134 строки
4.2 KiB
C++
134 строки
4.2 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 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/. */
|
|
|
|
#include "WebSocketFrame.h"
|
|
|
|
#include "ErrorList.h"
|
|
#include "MainThreadUtils.h"
|
|
#include "chrome/common/ipc_message_utils.h"
|
|
#include "mozilla/Assertions.h"
|
|
#include "nsDOMNavigationTiming.h"
|
|
#include "nsSocketTransportService2.h"
|
|
#include "nscore.h"
|
|
#include "prtime.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_INTERFACE_MAP_BEGIN(WebSocketFrame)
|
|
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIWebSocketFrame)
|
|
NS_INTERFACE_MAP_ENTRY(nsIWebSocketFrame)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
NS_IMPL_ADDREF(WebSocketFrame)
|
|
NS_IMPL_RELEASE(WebSocketFrame)
|
|
|
|
WebSocketFrame::WebSocketFrame(const WebSocketFrameData& aData)
|
|
: mData(aData) {}
|
|
|
|
WebSocketFrame::WebSocketFrame(bool aFinBit, bool aRsvBit1, bool aRsvBit2,
|
|
bool aRsvBit3, uint8_t aOpCode, bool aMaskBit,
|
|
uint32_t aMask, const nsCString& aPayload)
|
|
: mData(PR_Now(), aFinBit, aRsvBit1, aRsvBit2, aRsvBit3, aOpCode, aMaskBit,
|
|
aMask, aPayload) {
|
|
// This could be called on the background thread when socket process is
|
|
// enabled.
|
|
mData.mTimeStamp = PR_Now();
|
|
}
|
|
|
|
#define WSF_GETTER(method, value, type) \
|
|
NS_IMETHODIMP \
|
|
WebSocketFrame::method(type* aValue) { \
|
|
MOZ_ASSERT(NS_IsMainThread()); \
|
|
if (!aValue) { \
|
|
return NS_ERROR_FAILURE; \
|
|
} \
|
|
*aValue = value; \
|
|
return NS_OK; \
|
|
}
|
|
|
|
WSF_GETTER(GetTimeStamp, mData.mTimeStamp, DOMHighResTimeStamp);
|
|
WSF_GETTER(GetFinBit, mData.mFinBit, bool);
|
|
WSF_GETTER(GetRsvBit1, mData.mRsvBit1, bool);
|
|
WSF_GETTER(GetRsvBit2, mData.mRsvBit2, bool);
|
|
WSF_GETTER(GetRsvBit3, mData.mRsvBit3, bool);
|
|
WSF_GETTER(GetOpCode, mData.mOpCode, uint16_t);
|
|
WSF_GETTER(GetMaskBit, mData.mMaskBit, bool);
|
|
WSF_GETTER(GetMask, mData.mMask, uint32_t);
|
|
|
|
#undef WSF_GETTER
|
|
|
|
NS_IMETHODIMP
|
|
WebSocketFrame::GetPayload(nsACString& aValue) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
aValue = mData.mPayload;
|
|
return NS_OK;
|
|
}
|
|
|
|
WebSocketFrameData::WebSocketFrameData()
|
|
: mFinBit(false),
|
|
mRsvBit1(false),
|
|
mRsvBit2(false),
|
|
mRsvBit3(false),
|
|
mMaskBit(false) {}
|
|
|
|
WebSocketFrameData::WebSocketFrameData(DOMHighResTimeStamp aTimeStamp,
|
|
bool aFinBit, bool aRsvBit1,
|
|
bool aRsvBit2, bool aRsvBit3,
|
|
uint8_t aOpCode, bool aMaskBit,
|
|
uint32_t aMask,
|
|
const nsCString& aPayload)
|
|
: mTimeStamp(aTimeStamp),
|
|
mFinBit(aFinBit),
|
|
mRsvBit1(aRsvBit1),
|
|
mRsvBit2(aRsvBit2),
|
|
mRsvBit3(aRsvBit3),
|
|
mMaskBit(aMaskBit),
|
|
mOpCode(aOpCode),
|
|
mMask(aMask),
|
|
mPayload(aPayload) {}
|
|
|
|
void WebSocketFrameData::WriteIPCParams(IPC::MessageWriter* aWriter) const {
|
|
WriteParam(aWriter, mTimeStamp);
|
|
WriteParam(aWriter, mFinBit);
|
|
WriteParam(aWriter, mRsvBit1);
|
|
WriteParam(aWriter, mRsvBit2);
|
|
WriteParam(aWriter, mRsvBit3);
|
|
WriteParam(aWriter, mMaskBit);
|
|
WriteParam(aWriter, mOpCode);
|
|
WriteParam(aWriter, mMask);
|
|
WriteParam(aWriter, mPayload);
|
|
}
|
|
|
|
bool WebSocketFrameData::ReadIPCParams(IPC::MessageReader* aReader) {
|
|
if (!ReadParam(aReader, &mTimeStamp)) {
|
|
return false;
|
|
}
|
|
|
|
#define ReadParamHelper(x) \
|
|
{ \
|
|
bool bit; \
|
|
if (!ReadParam(aReader, &bit)) { \
|
|
return false; \
|
|
} \
|
|
(x) = bit; \
|
|
}
|
|
|
|
ReadParamHelper(mFinBit);
|
|
ReadParamHelper(mRsvBit1);
|
|
ReadParamHelper(mRsvBit2);
|
|
ReadParamHelper(mRsvBit3);
|
|
ReadParamHelper(mMaskBit);
|
|
|
|
#undef ReadParamHelper
|
|
|
|
return ReadParam(aReader, &mOpCode) && ReadParam(aReader, &mMask) &&
|
|
ReadParam(aReader, &mPayload);
|
|
}
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|