dom/file/ipc/IPCBlobInputStreamChild.h
author Andrea Marchesini <amarchesini@mozilla.com>
Mon, 11 Sep 2017 17:29:15 +0200
changeset 429573 d285843321b58f1b76afaaad3842cc6e42385df6
parent 412323 351a4d47496ecc6d03585474d89f48b602b007e7
child 454154 68a1d607cbc487975686cd7e5e3fbd10e76004c5
permissions -rw-r--r--
Bug 1397645 - Optimize IPCBlobInputStream slicing with the introduction of nsICloneableInputStreamWithRange, r=asuth

/* -*- 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_ipc_IPCBlobInputStreamChild_h
#define mozilla_dom_ipc_IPCBlobInputStreamChild_h

#include "mozilla/ipc/PIPCBlobInputStreamChild.h"
#include "mozilla/Mutex.h"
#include "mozilla/UniquePtr.h"
#include "nsIThread.h"
#include "nsTArray.h"

namespace mozilla {
namespace dom {

namespace workers {
class WorkerHolder;
}

class IPCBlobInputStream;

class IPCBlobInputStreamChild final
  : public mozilla::ipc::PIPCBlobInputStreamChild
{
public:
  enum ActorState
  {
    // The actor is connected via IPDL to the parent.
    eActive,

    // The actor is disconnected.
    eInactive,

    // The actor is waiting to be disconnected. Once it has been disconnected,
    // it will be reactivated on the DOM-File thread.
    eActiveMigrating,

    // The actor has been disconnected and it's waiting to be connected on the
    // DOM-File thread.
    eInactiveMigrating,
  };

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(IPCBlobInputStreamChild)

  IPCBlobInputStreamChild(const nsID& aID, uint64_t aSize);

  void
  ActorDestroy(IProtocol::ActorDestroyReason aReason) override;

  ActorState
  State();

  already_AddRefed<IPCBlobInputStream>
  CreateStream();

  void
  ForgetStream(IPCBlobInputStream* aStream);

  const nsID&
  ID() const
  {
    return mID;
  }

  uint64_t
  Size() const
  {
    return mSize;
  }

  void
  StreamNeeded(IPCBlobInputStream* aStream,
               nsIEventTarget* aEventTarget);

  mozilla::ipc::IPCResult
  RecvStreamReady(const OptionalIPCStream& aStream) override;

  void
  Shutdown();

  void
  Migrated();

private:
  ~IPCBlobInputStreamChild();

  // Raw pointers because these streams keep this actor alive. When the last
  // stream is unregister, the actor will be deleted. This list is protected by
  // mutex.
  nsTArray<IPCBlobInputStream*> mStreams;

  // This mutex protects mStreams because that can be touched in any thread.
  Mutex mMutex;

  const nsID mID;
  const uint64_t mSize;

  ActorState  mState;

  // This struct and the array are used for creating streams when needed.
  struct PendingOperation
  {
    RefPtr<IPCBlobInputStream> mStream;
    nsCOMPtr<nsIEventTarget> mEventTarget;
  };
  nsTArray<PendingOperation> mPendingOperations;

  nsCOMPtr<nsISerialEventTarget> mOwningEventTarget;

  UniquePtr<workers::WorkerHolder> mWorkerHolder;
};

} // namespace dom
} // namespace mozilla

#endif // mozilla_dom_ipc_IPCBlobInputStreamChild_h