summaryrefslogtreecommitdiffstats
path: root/ipc/unixsocket/ConnectionOrientedSocket.h
blob: a1236ec4ab94cd655a17768b324479666254209b (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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/* -*- Mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; tab-width: 40 -*- */
/* vim: set ts=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_ipc_ConnectionOrientedSocket_h
#define mozilla_ipc_ConnectionOrientedSocket_h

#include <sys/socket.h>
#include "DataSocket.h"
#include "mozilla/ipc/UnixSocketWatcher.h"

class MessageLoop;

namespace mozilla {
namespace ipc {

class UnixSocketConnector;

/*
 * |ConnectionOrientedSocketIO| and |ConnectionOrientedSocket| define
 * interfaces for implementing stream sockets on I/O and consumer thread.
 * |ListenSocket| uses these classes to handle accepted sockets.
 */

class ConnectionOrientedSocketIO
  : public DataSocketIO
  , public UnixSocketWatcher
{
public:
  virtual ~ConnectionOrientedSocketIO();

  nsresult Accept(int aFd,
                  const struct sockaddr* aAddress,
                  socklen_t aAddressLength);

  nsresult Connect();

  void Send(UnixSocketIOBuffer* aBuffer);

  // Methods for |UnixSocketWatcher|
  //

  void OnSocketCanReceiveWithoutBlocking() final;
  void OnSocketCanSendWithoutBlocking() final;

  void OnListening() final;
  void OnConnected() final;
  void OnError(const char* aFunction, int aErrno) final;

protected:
  /**
   * Constructs an instance of |ConnectionOrientedSocketIO|
   *
   * @param aConsumerLoop The socket's consumer thread.
   * @param aIOLoop The socket's I/O loop.
   * @param aFd The socket file descriptor.
   * @param aConnectionStatus The connection status for |aFd|.
   * @param aConnector Connector object for socket-type-specific methods.
   */
  ConnectionOrientedSocketIO(MessageLoop* aConsumerLoop,
                             MessageLoop* aIOLoop,
                             int aFd, ConnectionStatus aConnectionStatus,
                             UnixSocketConnector* aConnector);

  /**
   * Constructs an instance of |ConnectionOrientedSocketIO|
   *
   * @param aConsumerLoop The socket's consumer thread.
   * @param aIOLoop The socket's I/O loop.
   * @param aConnector Connector object for socket-type-specific methods.
   */
  ConnectionOrientedSocketIO(MessageLoop* aConsumerLoop,
                             MessageLoop* aIOLoop,
                             UnixSocketConnector* aConnector);

private:
  /**
   * Connector object used to create the connection we are currently using.
   */
  UniquePtr<UnixSocketConnector> mConnector;

  /**
   * Number of valid bytes in |mPeerAddress|.
   */
  socklen_t mPeerAddressLength;

  /**
   * Address of the socket's current peer.
   */
  struct sockaddr_storage mPeerAddress;
};

class ConnectionOrientedSocket : public DataSocket
{
public:
  /**
   * Prepares an instance of |ConnectionOrientedSocket| in DISCONNECTED
   * state for accepting a connection. Consumer-thread only.
   *
   * @param aConnector The new connector object, owned by the
   *                   connection-oriented socket.
   * @param aConsumerLoop The socket's consumer thread.
   * @param aIOLoop The socket's I/O thread.
   * @param[out] aIO, Returns an instance of |ConnectionOrientedSocketIO|.
   * @return NS_OK on success, or an XPCOM error code otherwise.
   */
  virtual nsresult PrepareAccept(UnixSocketConnector* aConnector,
                                 MessageLoop* aConsumerLoop,
                                 MessageLoop* aIOLoop,
                                 ConnectionOrientedSocketIO*& aIO) = 0;

protected:
  ConnectionOrientedSocket();
  virtual ~ConnectionOrientedSocket();
};

}
}

#endif // mozilla_ipc_ConnectionOrientedSocket