1 /*
2  *  Copyright 2004 The WebRTC Project Authors. All rights reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef WEBRTC_BASE_SOCKET_H__
12 #define WEBRTC_BASE_SOCKET_H__
13 
14 #include <errno.h>
15 
16 #if defined(WEBRTC_POSIX)
17 #include <sys/types.h>
18 #include <sys/socket.h>
19 #include <arpa/inet.h>
20 #include <netinet/in.h>
21 #define SOCKET_EACCES EACCES
22 #endif
23 
24 #if defined(WEBRTC_WIN)
25 #include "webrtc/base/win32.h"
26 #endif
27 
28 #include "webrtc/base/basictypes.h"
29 #include "webrtc/base/socketaddress.h"
30 
31 // Rather than converting errors into a private namespace,
32 // Reuse the POSIX socket api errors. Note this depends on
33 // Win32 compatibility.
34 
35 #if defined(WEBRTC_WIN)
36 #undef EWOULDBLOCK  // Remove errno.h's definition for each macro below.
37 #define EWOULDBLOCK WSAEWOULDBLOCK
38 #undef EINPROGRESS
39 #define EINPROGRESS WSAEINPROGRESS
40 #undef EALREADY
41 #define EALREADY WSAEALREADY
42 #undef ENOTSOCK
43 #define ENOTSOCK WSAENOTSOCK
44 #undef EDESTADDRREQ
45 #define EDESTADDRREQ WSAEDESTADDRREQ
46 #undef EMSGSIZE
47 #define EMSGSIZE WSAEMSGSIZE
48 #undef EPROTOTYPE
49 #define EPROTOTYPE WSAEPROTOTYPE
50 #undef ENOPROTOOPT
51 #define ENOPROTOOPT WSAENOPROTOOPT
52 #undef EPROTONOSUPPORT
53 #define EPROTONOSUPPORT WSAEPROTONOSUPPORT
54 #undef ESOCKTNOSUPPORT
55 #define ESOCKTNOSUPPORT WSAESOCKTNOSUPPORT
56 #undef EOPNOTSUPP
57 #define EOPNOTSUPP WSAEOPNOTSUPP
58 #undef EPFNOSUPPORT
59 #define EPFNOSUPPORT WSAEPFNOSUPPORT
60 #undef EAFNOSUPPORT
61 #define EAFNOSUPPORT WSAEAFNOSUPPORT
62 #undef EADDRINUSE
63 #define EADDRINUSE WSAEADDRINUSE
64 #undef EADDRNOTAVAIL
65 #define EADDRNOTAVAIL WSAEADDRNOTAVAIL
66 #undef ENETDOWN
67 #define ENETDOWN WSAENETDOWN
68 #undef ENETUNREACH
69 #define ENETUNREACH WSAENETUNREACH
70 #undef ENETRESET
71 #define ENETRESET WSAENETRESET
72 #undef ECONNABORTED
73 #define ECONNABORTED WSAECONNABORTED
74 #undef ECONNRESET
75 #define ECONNRESET WSAECONNRESET
76 #undef ENOBUFS
77 #define ENOBUFS WSAENOBUFS
78 #undef EISCONN
79 #define EISCONN WSAEISCONN
80 #undef ENOTCONN
81 #define ENOTCONN WSAENOTCONN
82 #undef ESHUTDOWN
83 #define ESHUTDOWN WSAESHUTDOWN
84 #undef ETOOMANYREFS
85 #define ETOOMANYREFS WSAETOOMANYREFS
86 #undef ETIMEDOUT
87 #define ETIMEDOUT WSAETIMEDOUT
88 #undef ECONNREFUSED
89 #define ECONNREFUSED WSAECONNREFUSED
90 #undef ELOOP
91 #define ELOOP WSAELOOP
92 #undef ENAMETOOLONG
93 #define ENAMETOOLONG WSAENAMETOOLONG
94 #undef EHOSTDOWN
95 #define EHOSTDOWN WSAEHOSTDOWN
96 #undef EHOSTUNREACH
97 #define EHOSTUNREACH WSAEHOSTUNREACH
98 #undef ENOTEMPTY
99 #define ENOTEMPTY WSAENOTEMPTY
100 #undef EPROCLIM
101 #define EPROCLIM WSAEPROCLIM
102 #undef EUSERS
103 #define EUSERS WSAEUSERS
104 #undef EDQUOT
105 #define EDQUOT WSAEDQUOT
106 #undef ESTALE
107 #define ESTALE WSAESTALE
108 #undef EREMOTE
109 #define EREMOTE WSAEREMOTE
110 #undef EACCES
111 #define SOCKET_EACCES WSAEACCES
112 #endif  // WEBRTC_WIN
113 
114 #if defined(WEBRTC_POSIX)
115 #define INVALID_SOCKET (-1)
116 #define SOCKET_ERROR (-1)
117 #define closesocket(s) close(s)
118 #endif  // WEBRTC_POSIX
119 
120 namespace rtc {
121 
IsBlockingError(int e)122 inline bool IsBlockingError(int e) {
123   return (e == EWOULDBLOCK) || (e == EAGAIN) || (e == EINPROGRESS);
124 }
125 
126 // General interface for the socket implementations of various networks.  The
127 // methods match those of normal UNIX sockets very closely.
128 class Socket {
129  public:
~Socket()130   virtual ~Socket() {}
131 
132   // Returns the address to which the socket is bound.  If the socket is not
133   // bound, then the any-address is returned.
134   virtual SocketAddress GetLocalAddress() const = 0;
135 
136   // Returns the address to which the socket is connected.  If the socket is
137   // not connected, then the any-address is returned.
138   virtual SocketAddress GetRemoteAddress() const = 0;
139 
140   virtual int Bind(const SocketAddress& addr) = 0;
141   virtual int Connect(const SocketAddress& addr) = 0;
142   virtual int Send(const void *pv, size_t cb) = 0;
143   virtual int SendTo(const void *pv, size_t cb, const SocketAddress& addr) = 0;
144   virtual int Recv(void *pv, size_t cb) = 0;
145   virtual int RecvFrom(void *pv, size_t cb, SocketAddress *paddr) = 0;
146   virtual int Listen(int backlog) = 0;
147   virtual Socket *Accept(SocketAddress *paddr) = 0;
148   virtual int Close() = 0;
149   virtual int GetError() const = 0;
150   virtual void SetError(int error) = 0;
IsBlocking()151   inline bool IsBlocking() const { return IsBlockingError(GetError()); }
152 
153   enum ConnState {
154     CS_CLOSED,
155     CS_CONNECTING,
156     CS_CONNECTED
157   };
158   virtual ConnState GetState() const = 0;
159 
160   // Fills in the given uint16 with the current estimate of the MTU along the
161   // path to the address to which this socket is connected. NOTE: This method
162   // can block for up to 10 seconds on Windows.
163   virtual int EstimateMTU(uint16* mtu) = 0;
164 
165   enum Option {
166     OPT_DONTFRAGMENT,
167     OPT_RCVBUF,      // receive buffer size
168     OPT_SNDBUF,      // send buffer size
169     OPT_NODELAY,     // whether Nagle algorithm is enabled
170     OPT_IPV6_V6ONLY, // Whether the socket is IPv6 only.
171     OPT_DSCP,        // DSCP code
172     OPT_RTP_SENDTIME_EXTN_ID,  // This is a non-traditional socket option param.
173                                // This is specific to libjingle and will be used
174                                // if SendTime option is needed at socket level.
175   };
176   virtual int GetOption(Option opt, int* value) = 0;
177   virtual int SetOption(Option opt, int value) = 0;
178 
179  protected:
Socket()180   Socket() {}
181 
182  private:
183   DISALLOW_EVIL_CONSTRUCTORS(Socket);
184 };
185 
186 }  // namespace rtc
187 
188 #endif  // WEBRTC_BASE_SOCKET_H__
189