dom/network/TCPSocketChild.h
author Emilio Cobos Álvarez <emilio@crisal.io>
Tue, 09 Apr 2019 00:11:03 +0000
changeset 468475 05cee7dc3b5ca0745802c80a27f13c2dc4638ebe
parent 457460 92091d537161e4df0986113e26762ed801190dfa
child 482204 4bb011651a380c34cd6d9834e4e582aba4febb1d
permissions -rw-r--r--
Bug 1542762 - Use an explicit stack to measure rule tree memory usage. r=heycam A patch of mine that makes us measure the rule tree more often triggers this. Differential Revision: https://phabricator.services.mozilla.com/D26595

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

#ifndef mozilla_dom_TCPSocketChild_h
#define mozilla_dom_TCPSocketChild_h

#include "mozilla/dom/TypedArray.h"
#include "mozilla/net/PTCPSocketChild.h"
#include "nsCycleCollectionParticipant.h"
#include "nsCOMPtr.h"
#include "js/TypeDecls.h"

class nsITCPSocketCallback;

namespace IPC {
bool DeserializeArrayBuffer(JSContext* cx,
                            const InfallibleTArray<uint8_t>& aBuffer,
                            JS::MutableHandle<JS::Value> aVal);
}

namespace mozilla {
namespace dom {

class TCPSocket;

class TCPSocketChildBase : public nsISupports {
 public:
  NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(TCPSocketChildBase)
  NS_DECL_CYCLE_COLLECTING_ISUPPORTS

  void AddIPDLReference();
  void ReleaseIPDLReference();

 protected:
  TCPSocketChildBase();
  virtual ~TCPSocketChildBase();

  nsCOMPtr<nsITCPSocketCallback> mSocket;
  bool mIPCOpen;
};

class TCPSocketChild : public mozilla::net::PTCPSocketChild,
                       public TCPSocketChildBase {
 public:
  NS_IMETHOD_(MozExternalRefCountType) Release() override;

  TCPSocketChild(const nsAString& aHost, const uint16_t& aPort,
                 nsIEventTarget* aTarget);
  ~TCPSocketChild();

  void SendOpen(nsITCPSocketCallback* aSocket, bool aUseSSL,
                bool aUseArrayBuffers);
  void SendWindowlessOpenBind(nsITCPSocketCallback* aSocket,
                              const nsACString& aRemoteHost,
                              uint16_t aRemotePort,
                              const nsACString& aLocalHost, uint16_t aLocalPort,
                              bool aUseSSL, bool aUseRealtimeOptions);
  NS_IMETHOD SendSendArray(nsTArray<uint8_t>& aArray, uint32_t aTrackingNumber);
  void SendSend(const nsACString& aData, uint32_t aTrackingNumber);
  nsresult SendSend(const ArrayBuffer& aData, uint32_t aByteOffset,
                    uint32_t aByteLength, uint32_t aTrackingNumber);
  void SendSendArray(nsTArray<uint8_t>* arr, uint32_t trackingNumber);
  void SetSocket(TCPSocket* aSocket);

  void GetHost(nsAString& aHost);
  void GetPort(uint16_t* aPort);

  mozilla::ipc::IPCResult RecvCallback(const nsString& aType,
                                       const CallbackData& aData,
                                       const uint32_t& aReadyState);
  mozilla::ipc::IPCResult RecvRequestDelete();
  mozilla::ipc::IPCResult RecvUpdateBufferedAmount(
      const uint32_t& aBufferred, const uint32_t& aTrackingNumber);
  nsresult SetFilterName(const nsACString& aFilterName);

 private:
  nsString mHost;
  uint16_t mPort;
  nsCString mFilterName;
  nsCOMPtr<nsIEventTarget> mIPCEventTarget;
};

}  // namespace dom
}  // namespace mozilla

#endif