/* 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