summaryrefslogtreecommitdiffstats
path: root/netwerk/protocol/websocket/IPCTransportProvider.cpp
blob: 356cf2d57cee969148a8c5ab39e13571a4259fc6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set sw=2 ts=8 et ft=cpp : */
/* 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 "mozilla/net/IPCTransportProvider.h"

#include "nsISocketTransport.h"
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"

namespace mozilla {
namespace net {

NS_IMPL_ISUPPORTS(TransportProviderParent,
                  nsITransportProvider,
                  nsIHttpUpgradeListener)

TransportProviderParent::TransportProviderParent()
{
  MOZ_COUNT_CTOR(TransportProviderParent);
}

TransportProviderParent::~TransportProviderParent()
{
  MOZ_COUNT_DTOR(TransportProviderParent);
}

NS_IMETHODIMP
TransportProviderParent::SetListener(nsIHttpUpgradeListener* aListener)
{
  MOZ_ASSERT(aListener);
  mListener = aListener;

  MaybeNotify();

  return NS_OK;
}

NS_IMETHODIMP
TransportProviderParent::GetIPCChild(mozilla::net::PTransportProviderChild** aChild)
{
  MOZ_CRASH("Don't call this in parent process");
  *aChild = nullptr;
  return NS_OK;
}

NS_IMETHODIMP
TransportProviderParent::OnTransportAvailable(nsISocketTransport* aTransport,
                                              nsIAsyncInputStream* aSocketIn,
                                              nsIAsyncOutputStream* aSocketOut)
{
  MOZ_ASSERT(aTransport && aSocketOut && aSocketOut);
  mTransport = aTransport;
  mSocketIn = aSocketIn;
  mSocketOut = aSocketOut;

  MaybeNotify();

  return NS_OK;
}

void
TransportProviderParent::MaybeNotify()
{
  if (!mListener || !mTransport) {
    return;
  }

  mListener->OnTransportAvailable(mTransport, mSocketIn, mSocketOut);
}


NS_IMPL_ISUPPORTS(TransportProviderChild,
                  nsITransportProvider)

TransportProviderChild::TransportProviderChild()
{
  MOZ_COUNT_CTOR(TransportProviderChild);
}

TransportProviderChild::~TransportProviderChild()
{
  MOZ_COUNT_DTOR(TransportProviderChild);
  Send__delete__(this);
}

NS_IMETHODIMP
TransportProviderChild::SetListener(nsIHttpUpgradeListener* aListener)
{
  MOZ_CRASH("Don't call this in child process");
  return NS_OK;
}

NS_IMETHODIMP
TransportProviderChild::GetIPCChild(mozilla::net::PTransportProviderChild** aChild)
{
  *aChild = this;
  return NS_OK;
}

} // net
} // mozilla