Mercurial > code
annotate C++/Socket.cpp @ 246:9cfa6fbc9c03
Socket: add operators for SocketAddress
author | David Demelier <markand@malikania.fr> |
---|---|
date | Mon, 29 Sep 2014 12:57:43 +0200 |
parents | 3c12f0e8bbb9 |
children | 806dbb6011c7 |
rev | line source |
---|---|
170 | 1 /* |
2 * Socket.cpp -- portable C++ socket wrappers | |
3 * | |
4 * Copyright (c) 2013, David Demelier <markand@malikania.fr> | |
5 * | |
6 * Permission to use, copy, modify, and/or distribute this software for any | |
7 * purpose with or without fee is hereby granted, provided that the above | |
8 * copyright notice and this permission notice appear in all copies. | |
9 * | |
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES | |
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF | |
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR | |
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF | |
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |
17 */ | |
18 | |
176
ba62ab219fc4
add cstring for strerror
David Demelier <markand@malikania.fr>
parents:
175
diff
changeset
|
19 #include <cstring> |
170 | 20 |
21 #include "Socket.h" | |
22 #include "SocketAddress.h" | |
23 | |
24 /* -------------------------------------------------------- | |
25 * SocketError implementation | |
26 * -------------------------------------------------------- */ | |
27 | |
28 SocketError::SocketError(const std::string &error) | |
29 { | |
30 m_error = error; | |
31 } | |
32 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
33 const char *SocketError::what() const noexcept |
170 | 34 { |
35 return m_error.c_str(); | |
36 } | |
37 | |
38 /* -------------------------------------------------------- | |
39 * Socket implementation | |
40 * -------------------------------------------------------- */ | |
41 | |
42 void Socket::init() | |
43 { | |
44 #if defined(_WIN32) | |
45 WSADATA wsa; | |
46 WSAStartup(MAKEWORD(2, 2), &wsa); | |
47 #endif | |
48 } | |
49 | |
50 /* -------------------------------------------------------- | |
51 * System dependent code | |
52 * -------------------------------------------------------- */ | |
53 | |
54 #if defined(_WIN32) | |
55 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
56 std::string Socket::syserror() |
170 | 57 { |
58 LPSTR str = nullptr; | |
188
ce3e1c3d6fed
Update sockets to style and using instead of typedef
David Demelier <markand@malikania.fr>
parents:
176
diff
changeset
|
59 std::string errmsg = "Unknown error"; |
170 | 60 |
61 FormatMessageA( | |
62 FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, | |
63 NULL, | |
64 WSAGetLastError(), | |
65 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), | |
66 (LPSTR)&str, 0, NULL); | |
67 | |
68 | |
188
ce3e1c3d6fed
Update sockets to style and using instead of typedef
David Demelier <markand@malikania.fr>
parents:
176
diff
changeset
|
69 if (str) { |
170 | 70 errmsg = string(str); |
71 LocalFree(str); | |
72 } | |
73 | |
74 return errmsg; | |
75 } | |
76 | |
77 #else | |
78 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
79 #include <cerrno> |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
80 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
81 std::string Socket::syserror() |
170 | 82 { |
83 return strerror(errno); | |
84 } | |
85 | |
86 #endif | |
87 | |
88 void Socket::finish() | |
89 { | |
90 #if defined(_WIN32) | |
91 WSACleanup(); | |
92 #endif | |
93 } | |
94 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
95 /* -------------------------------------------------------- |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
96 * Standard clear implementation |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
97 * -------------------------------------------------------- */ |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
98 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
99 class Standard final : public SocketInterface { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
100 public: |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
101 void bind(Socket &s, const SocketAddress &address) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
102 void close(Socket &s) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
103 void connect(Socket &s, const SocketAddress &address) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
104 Socket accept(Socket &s, SocketAddress &info) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
105 void listen(Socket &s, int max) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
106 unsigned recv(Socket &s, void *data, unsigned len) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
107 unsigned recvfrom(Socket &s, void *data, unsigned len, SocketAddress &info) override ; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
108 unsigned send(Socket &s, const void *data, unsigned len) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
109 unsigned sendto(Socket &s, const void *data, unsigned len, const SocketAddress &info) override; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
110 }; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
111 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
112 void Standard::bind(Socket &s, const SocketAddress &addr) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
113 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
114 auto &sa = addr.address(); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
115 auto addrlen = addr.length(); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
116 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
117 if (::bind(s.handle(), (sockaddr *)&sa, addrlen) == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
118 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
119 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
120 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
121 void Standard::connect(Socket &s, const SocketAddress &addr) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
122 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
123 auto &sa = addr.address(); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
124 auto addrlen = addr.length(); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
125 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
126 if (::connect(s.handle(), (sockaddr *)&sa, addrlen) == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
127 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
128 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
129 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
130 Socket Standard::accept(Socket &s, SocketAddress &info) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
131 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
132 Socket c; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
133 Socket::Handle handle; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
134 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
135 // Store the information |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
136 sockaddr_storage address; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
137 socklen_t addrlen; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
138 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
139 addrlen = sizeof (sockaddr_storage); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
140 handle = ::accept(s.handle(), (sockaddr *)&address, &addrlen); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
141 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
142 if (handle == INVALID_SOCKET) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
143 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
144 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
145 // Usually accept works only with SOCK_STREAM |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
146 info = SocketAddress(address, addrlen); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
147 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
148 return Socket(handle, address.ss_family, s.type(), s.protocol()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
149 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
150 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
151 void Standard::listen(Socket &s, int max) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
152 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
153 if (::listen(s.handle(), max) == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
154 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
155 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
156 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
157 unsigned Standard::recv(Socket &s, void *data, unsigned dataLen) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
158 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
159 int nbread; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
160 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
161 nbread = ::recv(s.handle(), (Socket::Arg)data, dataLen, 0); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
162 if (nbread == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
163 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
164 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
165 return (unsigned)nbread; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
166 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
167 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
168 unsigned Standard::send(Socket &s, const void *data, unsigned dataLen) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
169 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
170 int nbsent; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
171 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
172 nbsent = ::send(s.handle(), (Socket::ConstArg)data, dataLen, 0); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
173 if (nbsent == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
174 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
175 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
176 return (unsigned)nbsent; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
177 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
178 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
179 unsigned Standard::recvfrom(Socket &s, void *data, unsigned dataLen, SocketAddress &info) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
180 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
181 int nbread; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
182 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
183 // Store information |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
184 sockaddr_storage address; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
185 socklen_t addrlen; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
186 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
187 addrlen = sizeof (struct sockaddr_storage); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
188 nbread = ::recvfrom(s.handle(), (Socket::Arg)data, dataLen, 0, (sockaddr *)&address, &addrlen); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
189 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
190 if (nbread == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
191 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
192 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
193 return (unsigned)nbread; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
194 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
195 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
196 unsigned Standard::sendto(Socket &s, const void *data, unsigned dataLen, const SocketAddress &info) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
197 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
198 int nbsent; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
199 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
200 nbsent = ::sendto(s.handle(), (Socket::ConstArg)data, dataLen, 0, (const sockaddr *)&info.address(), info.length()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
201 if (nbsent == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
202 throw SocketError(Socket::syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
203 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
204 return (unsigned)nbsent; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
205 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
206 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
207 void Standard::close(Socket &s) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
208 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
209 (void)closesocket(s.handle()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
210 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
211 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
212 /* -------------------------------------------------------- |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
213 * Socket code |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
214 * -------------------------------------------------------- */ |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
215 |
170 | 216 Socket::Socket() |
175 | 217 : m_domain(0) |
218 , m_type(0) | |
219 , m_protocol(0) | |
170 | 220 { |
221 } | |
222 | |
223 Socket::Socket(int domain, int type, int protocol) | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
224 : m_interface(std::make_unique<Standard>()) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
225 , m_domain(domain) |
175 | 226 , m_type(type) |
227 , m_protocol(protocol) | |
170 | 228 { |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
229 m_handle = socket(domain, type, protocol); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
230 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
231 if (m_handle == INVALID_SOCKET) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
232 throw SocketError(syserror()); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
233 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
234 m_open = true; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
235 } |
170 | 236 |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
237 Socket::Socket(Handle handle, int domain, int type, int protocol) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
238 : m_handle(handle) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
239 , m_domain(domain) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
240 , m_type(type) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
241 , m_protocol(protocol) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
242 , m_open(true) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
243 { |
170 | 244 } |
245 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
246 Socket::Socket(Socket &&other) noexcept |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
247 : m_interface(std::move(other.m_interface)) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
248 , m_handle(std::move(other.m_handle)) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
249 , m_domain(other.m_domain) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
250 , m_type(other.m_type) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
251 , m_protocol(other.m_protocol) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
252 , m_open(true) |
170 | 253 { |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
254 // invalidate other |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
255 other.m_handle = 0; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
256 other.m_open = false; |
170 | 257 } |
258 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
259 Socket &Socket::operator=(Socket &&other) noexcept |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
260 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
261 m_interface = std::move(other.m_interface); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
262 m_handle = std::move(other.m_handle); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
263 m_domain = other.m_domain; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
264 m_type = other.m_type; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
265 m_protocol = other.m_protocol; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
266 m_open = true; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
267 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
268 // invalidate other |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
269 other.m_handle = 0; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
270 other.m_open = false; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
271 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
272 return *this; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
273 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
274 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
275 Socket::~Socket() |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
276 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
277 close(); |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
278 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
279 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
280 Socket::Handle Socket::handle() const |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
281 { |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
282 return m_handle; |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
283 } |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
284 |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
285 int Socket::domain() const |
175 | 286 { |
287 return m_domain; | |
288 } | |
289 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
290 int Socket::type() const |
175 | 291 { |
292 return m_type; | |
293 } | |
294 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
295 int Socket::protocol() const |
175 | 296 { |
297 return m_protocol; | |
298 } | |
299 | |
170 | 300 void Socket::set(int level, int name, const void *arg, unsigned argLen) |
301 { | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
302 if (setsockopt(m_handle, level, name, (Socket::ConstArg)arg, argLen) == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
303 throw SocketError(syserror()); |
170 | 304 } |
305 | |
306 void Socket::blockMode(bool block) | |
307 { | |
308 #if defined(O_NONBLOCK) && !defined(_WIN32) | |
309 int flags; | |
310 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
311 if ((flags = fcntl(m_handle, F_GETFL, 0)) == -1) |
170 | 312 flags = 0; |
313 | |
314 if (!block) | |
315 flags &= ~(O_NONBLOCK); | |
316 else | |
317 flags |= O_NONBLOCK; | |
318 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
319 if (fcntl(m_handle, F_SETFL, flags) == -1) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
320 throw SocketError(Socket::syserror()); |
170 | 321 #else |
322 unsigned long flags = (block) ? 0 : 1; | |
323 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
324 if (ioctlsocket(m_handle, FIONBIO, &flags) == SOCKET_ERROR) |
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
325 throw SocketError(Socket::syserror()); |
170 | 326 #endif |
327 } | |
328 | |
329 Socket Socket::accept() | |
330 { | |
331 SocketAddress dummy; | |
332 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
333 return m_interface->accept(*this, dummy); |
170 | 334 } |
335 | |
336 unsigned Socket::send(const std::string &message) | |
337 { | |
338 return Socket::send(message.c_str(), message.length()); | |
339 } | |
340 | |
341 unsigned Socket::recvfrom(void *data, unsigned dataLen) | |
342 { | |
343 SocketAddress dummy; | |
344 | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
345 return m_interface->recvfrom(*this, data, dataLen, dummy); |
170 | 346 } |
347 | |
348 unsigned Socket::sendto(const std::string &message, const SocketAddress &info) | |
349 { | |
350 return sendto(message.c_str(), message.length(), info); | |
351 } | |
352 | |
353 bool operator==(const Socket &s1, const Socket &s2) | |
354 { | |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
355 return s1.handle() == s2.handle(); |
170 | 356 } |
174
9f22bd478f21
Add operator less because widely used
David Demelier <markand@malikania.fr>
parents:
170
diff
changeset
|
357 |
175 | 358 bool operator<(const Socket &s1, const Socket &s2) |
174
9f22bd478f21
Add operator less because widely used
David Demelier <markand@malikania.fr>
parents:
170
diff
changeset
|
359 { |
245
3c12f0e8bbb9
Converter: add cstring missing
David Demelier <markand@malikania.fr>
parents:
236
diff
changeset
|
360 return s1.handle() < s2.handle(); |
174
9f22bd478f21
Add operator less because widely used
David Demelier <markand@malikania.fr>
parents:
170
diff
changeset
|
361 } |